YajieXu commited on
Commit
7b67e95
·
verified ·
1 Parent(s): 8601660

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -13,12 +13,10 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
13
  class GAIAAgent:
14
  def __init__(self):
15
  print("GAIAAgent with HfApiModel initialized.")
16
- self.model = HfApiModel(
17
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
18
- max_tokens=2096,
19
- temperature=0.5,
20
- custom_role_conversions=None,
21
- )
22
 
23
  def format_prompt(self, question: str, file_content: str = None) -> str:
24
  prompt = (
 
13
  class GAIAAgent:
14
  def __init__(self):
15
  print("GAIAAgent with HfApiModel initialized.")
16
+ self.model = gr.load(
17
+ "models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
18
+ provider="novita",
19
+ ).launch()
 
 
20
 
21
  def format_prompt(self, question: str, file_content: str = None) -> str:
22
  prompt = (