MyNameIsSimon commited on
Commit
130ba32
·
1 Parent(s): f59225d

added new model to model selection

Browse files
Files changed (1) hide show
  1. app.py +7 -1
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 = ["lab2-as/lora_model_gguf, Q4", "lab2-as/lora_model_no_quant_gguf, Q4"]
 
 
 
 
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,