skjaini commited on
Commit
0ca5145
·
verified ·
1 Parent(s): 84df946

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -46,6 +46,7 @@ def create_ai_news_crew():
46
  tools=[search_tool],
47
  llm=HuggingFaceHub(
48
  repo_id="deepseek-ai/DeepSeek-Coder-33B-Instruct",
 
49
  model_kwargs={"temperature": 0.5, "max_new_tokens": 1024, "repetition_penalty": 1.2},
50
  huggingfacehub_api_token=huggingfacehub_api_token,
51
  )
@@ -62,6 +63,7 @@ def create_ai_news_crew():
62
  allow_delegation=False,
63
  llm=HuggingFaceHub(
64
  repo_id="deepseek-ai/DeepSeek-Coder-33B-Instruct",
 
65
  model_kwargs={"temperature": 0.2, "max_new_tokens": 1024, "repetition_penalty": 1.2},
66
  huggingfacehub_api_token=huggingfacehub_api_token,
67
  )
@@ -76,7 +78,7 @@ def create_ai_news_crew():
76
  Return the titles and URLs of the most important articles.
77
  """,
78
  agent=researcher,
79
- expected_output="A list of titles and URLs of the most important AI news articles from yesterday." # Added expected_output
80
  )
81
 
82
  summarize_task = Task(
@@ -87,7 +89,7 @@ def create_ai_news_crew():
87
  Format the output using markdown for readability.
88
  """,
89
  agent=summarizer,
90
- expected_output="A concise daily AI news briefing in markdown format, no more than 500 words." # Added expected_output
91
  )
92
 
93
  # Create Crew
 
46
  tools=[search_tool],
47
  llm=HuggingFaceHub(
48
  repo_id="deepseek-ai/DeepSeek-Coder-33B-Instruct",
49
+ task="text-generation", # Explicitly set the task
50
  model_kwargs={"temperature": 0.5, "max_new_tokens": 1024, "repetition_penalty": 1.2},
51
  huggingfacehub_api_token=huggingfacehub_api_token,
52
  )
 
63
  allow_delegation=False,
64
  llm=HuggingFaceHub(
65
  repo_id="deepseek-ai/DeepSeek-Coder-33B-Instruct",
66
+ task="text-generation", # Explicitly set the task
67
  model_kwargs={"temperature": 0.2, "max_new_tokens": 1024, "repetition_penalty": 1.2},
68
  huggingfacehub_api_token=huggingfacehub_api_token,
69
  )
 
78
  Return the titles and URLs of the most important articles.
79
  """,
80
  agent=researcher,
81
+ expected_output="A list of titles and URLs of the most important AI news articles from yesterday."
82
  )
83
 
84
  summarize_task = Task(
 
89
  Format the output using markdown for readability.
90
  """,
91
  agent=summarizer,
92
+ expected_output="A concise daily AI news briefing in markdown format, no more than 500 words."
93
  )
94
 
95
  # Create Crew