Commit
·
130ba32
1
Parent(s):
f59225d
added new model to model selection
Browse files
app.py
CHANGED
|
@@ -63,6 +63,8 @@ class MyModel:
|
|
| 63 |
|
| 64 |
|
| 65 |
transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-base.en")
|
|
|
|
|
|
|
| 66 |
def transcribe(audio):
|
| 67 |
sr, y = audio
|
| 68 |
|
|
@@ -81,7 +83,11 @@ def transcribe(audio):
|
|
| 81 |
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
|
| 82 |
"""
|
| 83 |
my_model = MyModel()
|
| 84 |
-
model_choices = [
|
|
|
|
|
|
|
|
|
|
|
|
|
| 85 |
demo = CustomChatInterface(
|
| 86 |
my_model.respond,
|
| 87 |
transcriber=transcribe,
|
|
|
|
| 63 |
|
| 64 |
|
| 65 |
transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-base.en")
|
| 66 |
+
|
| 67 |
+
|
| 68 |
def transcribe(audio):
|
| 69 |
sr, y = audio
|
| 70 |
|
|
|
|
| 83 |
For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
|
| 84 |
"""
|
| 85 |
my_model = MyModel()
|
| 86 |
+
model_choices = [
|
| 87 |
+
"lab2-as/lora_model_gguf, Q4",
|
| 88 |
+
"lab2-as/lora_model_no_quant_gguf, Q4",
|
| 89 |
+
"lab2-as/lora_model_math_optimized_gguf, Q4",
|
| 90 |
+
]
|
| 91 |
demo = CustomChatInterface(
|
| 92 |
my_model.respond,
|
| 93 |
transcriber=transcribe,
|