add gpt-4o

app.py

@@ -89,7 +89,7 @@ def do( business_name: str, address: str):
     formatted_output = format_category( formatted_results)
 
     img = plot_wordcloud(formatted_results['formatted_evidence'].values[0])
-    return f"【參考資料】\n{formatted_results['formatted_evidence'].values[0]}", img, f"【辨識結果】\n{formatted_output}"
+    return f"【搜尋結果】\n{formatted_results['formatted_evidence'].values[0]}", img, f"【判斷結果】\n{formatted_output}"
 
 ## --- interface --- ##
 # outputs = [gr.Dataframe(row_count = (1, "dynamic"), col_count=(6,"dynamic"), label="output data", interactive=1)]

model.py

@@ -69,7 +69,7 @@ def llm( provider, model, system_prompt, user_content, delay:int = 0):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--provider", type=str, default='anthropic', help="openai or anthropic")
-    parser.add_argument("--model", type=str, default='claude-3-sonnet-20240229', help="Model name for the API", 
+    parser.add_argument("--model", type=str, default='gpt-4o', help="Model name for the API", 
                         choices = ["claude-3-sonnet-20240229", "claude-3-haiku-20240307", "gpt-3.5-turbo-0125", "gpt-4-0125-preview"])
     parser.add_argument("--classes", type=list, default=['小吃店', '日式料理(含居酒屋，串燒)', '火(鍋／爐)', '東南亞料理(不含日韓)', '海鮮熱炒',  '特色餐廳(含雞、鵝、牛、羊肉)', '傳統餐廳', '燒烤', '韓式料理(含火鍋，烤肉)', '西餐廳(含美式，義式，墨式)', '西餐廳(餐酒館、酒吧、飛鏢吧、pub、lounge bar)', '西餐廳(土耳其、漢堡、薯條、法式、歐式、印度)', '早餐'])
     parser.add_argument("--task", type=list, default='extract', choices=['extract', 'classify'])
@@ -130,4 +130,7 @@ if __name__ == "__main__":
         '''
         print(f"user_content -> {user_content}")
         resp = llm( args.provider, args.model, system_prompt, user_content)
-        print(resp)
+        print(resp)
+
+    else:
+        raise Exception("Invalid provider")