rohan13 commited on
Commit
0fb5725
·
1 Parent(s): 451ac96
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,11 +5,11 @@ import os, time
5
 
6
  from transformers import pipeline
7
 
8
- p = pipeline("automatic-speech-recognition")
9
 
10
  """Use text to call chat method from main.py"""
11
 
12
- models = ["GPT-3.5", "Flan UL2", "Flan T5", "Palm"]
13
 
14
  name = os.environ.get("name", "Rohan")
15
 
@@ -100,7 +100,7 @@ with gr.Blocks(theme=theme, title='Roar!') as demo:
100
 
101
  with gr.Row():
102
  # Create radio button to select model
103
- radio = gr.Radio(models, label="Choose a model", value="GPT-3.5", type="value")
104
  with gr.Row():
105
  with gr.Column(scale=0.6):
106
  txt = gr.Textbox(
 
5
 
6
  from transformers import pipeline
7
 
8
+ p = pipeline("automatic-speech-recognition", model="openai/whisper-base")
9
 
10
  """Use text to call chat method from main.py"""
11
 
12
+ models = ["GPT-3.5", "Flan UL2", "Flan T5", "GPT-4"]
13
 
14
  name = os.environ.get("name", "Rohan")
15
 
 
100
 
101
  with gr.Row():
102
  # Create radio button to select model
103
+ radio = gr.Radio(models, label="Choose a model", value="GPT-4", type="value")
104
  with gr.Row():
105
  with gr.Column(scale=0.6):
106
  txt = gr.Textbox(