You-shen commited on
Commit
6fa2204
·
verified ·
1 Parent(s): d33faff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -550,6 +550,8 @@ def SelectModel(option):
550
  global selected_model
551
  if option == "自研":
552
  selected_model = "gpt"
 
 
553
  elif option == "llama3-70b":
554
  selected_model = "llama3-70b"
555
  elif option == "llama3-8b":
@@ -558,7 +560,7 @@ def SelectModel(option):
558
  selected_model = "mistral"
559
 
560
 
561
- model_option = ["自研", "llama3-70b", "llama3-8b", "mistral"]
562
 
563
 
564
  def start(topic, image_summaries):
 
550
  global selected_model
551
  if option == "自研":
552
  selected_model = "gpt"
553
+ elif option == "gpt-4o":
554
+ selected_model = "gpt"
555
  elif option == "llama3-70b":
556
  selected_model = "llama3-70b"
557
  elif option == "llama3-8b":
 
560
  selected_model = "mistral"
561
 
562
 
563
+ model_option = ["自研", "gpt-4o","llama3-70b", "llama3-8b", "mistral"]
564
 
565
 
566
  def start(topic, image_summaries):