kz209 commited on
Commit
5e835e3
·
1 Parent(s): ebdc474

update models

Browse files
Files changed (1) hide show
  1. pages/summarization_example.py +1 -2
pages/summarization_example.py CHANGED
@@ -4,7 +4,6 @@ import gradio as gr
4
 
5
  from utils.model import Model
6
 
7
-
8
  model = Model()
9
 
10
  load_dotenv()
@@ -50,7 +49,7 @@ def create_summarization_interface():
50
 
51
  with gr.Row():
52
  example_dropdown = gr.Dropdown(choices=list(examples.keys()), label="Choose an example")
53
- model_dropdown = gr.Dropdown(choices=["gpt-3.5-turbo", "gpt-4o", "gpt-4"], label="Choose a model", value="gpt-3.5-turbo")
54
 
55
  input_text = gr.Textbox(label="Input Text", lines=10, placeholder="Enter text here...")
56
  submit_button = gr.Button("✨ Submit ✨")
 
4
 
5
  from utils.model import Model
6
 
 
7
  model = Model()
8
 
9
  load_dotenv()
 
49
 
50
  with gr.Row():
51
  example_dropdown = gr.Dropdown(choices=list(examples.keys()), label="Choose an example")
52
+ model_dropdown = gr.Dropdown(choices=["lmsys/vicuna-7b-v1.5", "tiiuae/falcon-7b-instruct"], label="Choose a model", value="gpt-3.5-turbo")
53
 
54
  input_text = gr.Textbox(label="Input Text", lines=10, placeholder="Enter text here...")
55
  submit_button = gr.Button("✨ Submit ✨")