gokaygokay commited on
Commit
25aca96
·
1 Parent(s): 4fa548e

fix models

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -50,7 +50,16 @@ def create_video_interface():
50
  value="Hugging Face",
51
  label="LLM Provider"
52
  )
53
- model = gr.Dropdown(label="Model", value="Qwen/Qwen2.5-72B-Instruct")
 
 
 
 
 
 
 
 
 
54
 
55
  generate_btn = gr.Button("Generate Video Prompt", variant="primary")
56
  output = gr.Textbox(label="Generated Prompt", lines=12, show_copy_button=True)
 
50
  value="Hugging Face",
51
  label="LLM Provider"
52
  )
53
+ model = gr.Dropdown(
54
+ choices=[
55
+ "Qwen/Qwen2.5-72B-Instruct",
56
+ "meta-llama/Meta-Llama-3.1-70B-Instruct",
57
+ "mistralai/Mixtral-8x7B-Instruct-v0.1",
58
+ "mistralai/Mistral-7B-Instruct-v0.3"
59
+ ],
60
+ value="Qwen/Qwen2.5-72B-Instruct",
61
+ label="Model"
62
+ )
63
 
64
  generate_btn = gr.Button("Generate Video Prompt", variant="primary")
65
  output = gr.Textbox(label="Generated Prompt", lines=12, show_copy_button=True)