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,
|