linpershey commited on
Commit
2547429
1 Parent(s): 8aa4241

add gpt-4o

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. model.py +5 -2
app.py CHANGED
@@ -89,7 +89,7 @@ def do( business_name: str, address: str):
89
  formatted_output = format_category( formatted_results)
90
 
91
  img = plot_wordcloud(formatted_results['formatted_evidence'].values[0])
92
- return f"【參考資料】\n{formatted_results['formatted_evidence'].values[0]}", img, f"【辨識結果】\n{formatted_output}"
93
 
94
  ## --- interface --- ##
95
  # outputs = [gr.Dataframe(row_count = (1, "dynamic"), col_count=(6,"dynamic"), label="output data", interactive=1)]
 
89
  formatted_output = format_category( formatted_results)
90
 
91
  img = plot_wordcloud(formatted_results['formatted_evidence'].values[0])
92
+ return f"【搜尋結果】\n{formatted_results['formatted_evidence'].values[0]}", img, f"【判斷結果】\n{formatted_output}"
93
 
94
  ## --- interface --- ##
95
  # outputs = [gr.Dataframe(row_count = (1, "dynamic"), col_count=(6,"dynamic"), label="output data", interactive=1)]
model.py CHANGED
@@ -69,7 +69,7 @@ def llm( provider, model, system_prompt, user_content, delay:int = 0):
69
  if __name__ == "__main__":
70
  parser = argparse.ArgumentParser()
71
  parser.add_argument("--provider", type=str, default='anthropic', help="openai or anthropic")
72
- parser.add_argument("--model", type=str, default='claude-3-sonnet-20240229', help="Model name for the API",
73
  choices = ["claude-3-sonnet-20240229", "claude-3-haiku-20240307", "gpt-3.5-turbo-0125", "gpt-4-0125-preview"])
74
  parser.add_argument("--classes", type=list, default=['小吃店', '日式料理(含居酒屋,串燒)', '火(鍋/爐)', '東南亞料理(不含日韓)', '海鮮熱炒', '特色餐廳(含雞、鵝、牛、羊肉)', '傳統餐廳', '燒烤', '韓式料理(含火鍋,烤肉)', '西餐廳(含美式,義式,墨式)', '西餐廳(餐酒館、酒吧、飛鏢吧、pub、lounge bar)', '西餐廳(土耳其、漢堡、薯條、法式、歐式、印度)', '早餐'])
75
  parser.add_argument("--task", type=list, default='extract', choices=['extract', 'classify'])
@@ -130,4 +130,7 @@ if __name__ == "__main__":
130
  '''
131
  print(f"user_content -> {user_content}")
132
  resp = llm( args.provider, args.model, system_prompt, user_content)
133
- print(resp)
 
 
 
 
69
  if __name__ == "__main__":
70
  parser = argparse.ArgumentParser()
71
  parser.add_argument("--provider", type=str, default='anthropic', help="openai or anthropic")
72
+ parser.add_argument("--model", type=str, default='gpt-4o', help="Model name for the API",
73
  choices = ["claude-3-sonnet-20240229", "claude-3-haiku-20240307", "gpt-3.5-turbo-0125", "gpt-4-0125-preview"])
74
  parser.add_argument("--classes", type=list, default=['小吃店', '日式料理(含居酒屋,串燒)', '火(鍋/爐)', '東南亞料理(不含日韓)', '海鮮熱炒', '特色餐廳(含雞、鵝、牛、羊肉)', '傳統餐廳', '燒烤', '韓式料理(含火鍋,烤肉)', '西餐廳(含美式,義式,墨式)', '西餐廳(餐酒館、酒吧、飛鏢吧、pub、lounge bar)', '西餐廳(土耳其、漢堡、薯條、法式、歐式、印度)', '早餐'])
75
  parser.add_argument("--task", type=list, default='extract', choices=['extract', 'classify'])
 
130
  '''
131
  print(f"user_content -> {user_content}")
132
  resp = llm( args.provider, args.model, system_prompt, user_content)
133
+ print(resp)
134
+
135
+ else:
136
+ raise Exception("Invalid provider")