Ajeya95 commited on
Commit
64af345
·
verified ·
1 Parent(s): c42c041

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -7,10 +7,10 @@ from tools.final_answer import FinalAnswerTool
7
  import json
8
  from Gradio_UI import GradioUI
9
  # Load model directly
10
- from transformers import AutoProcessor, AutoModel
11
 
12
- processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-Omni-3B")
13
- model = AutoModel.from_pretrained("Qwen/Qwen2.5-Omni-3B")
14
 
15
  duckduckgo_tool=DuckDuckGoSearchTool()
16
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -82,6 +82,7 @@ final_answer = FinalAnswerTool()
82
 
83
  model = HfApiModel(
84
  max_tokens=2096,
 
85
  temperature=0.5,
86
  model_id= 'Qwen/Qwen2.5-Omni-3B',# it is possible that this model may be overloaded
87
  custom_role_conversions=None,
 
7
  import json
8
  from Gradio_UI import GradioUI
9
  # Load model directly
10
+ # from transformers import AutoProcessor, AutoModel
11
 
12
+ # processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-Omni-3B")
13
+ # model = AutoModel.from_pretrained("Qwen/Qwen2.5-Omni-3B")
14
 
15
  duckduckgo_tool=DuckDuckGoSearchTool()
16
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
82
 
83
  model = HfApiModel(
84
  max_tokens=2096,
85
+
86
  temperature=0.5,
87
  model_id= 'Qwen/Qwen2.5-Omni-3B',# it is possible that this model may be overloaded
88
  custom_role_conversions=None,