Crackershoot commited on
Commit
3198449
·
verified ·
1 Parent(s): 03c2ed7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -13,11 +13,12 @@ import yaml
13
  from tools.final_answer import FinalAnswerTool
14
  from Gradio_UI import GradioUI
15
 
16
- model = LiteLLMModel(
17
- model_id = "ollama_chat/qwen2:7b", # Or try other Ollama-supported models
18
- api_base = "http://127.0.0.1:11434", # Default Ollama local server
19
- num_ctx = 8192,
20
- )
 
21
 
22
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
23
  @tool
 
13
  from tools.final_answer import FinalAnswerTool
14
  from Gradio_UI import GradioUI
15
 
16
+ # model = LiteLLMModel(
17
+ # model_id = "ollama_chat/qwen2:7b", # Or try other Ollama-supported models
18
+ # api_base = "http://127.0.0.1:11434", # Default Ollama local server
19
+ # num_ctx = 8192,
20
+ # )
21
+ model = LiteLLMModel(model_id="gpt-4")
22
 
23
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
24
  @tool