Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from crewai import Crew, Agent, Task, Process
 from langchain_core.tools import Tool
 from langchain_community.tools import DuckDuckGoSearchRun
-from langchain_community.llms import HuggingFaceEndpoint
+from langchain_community.llms import HuggingFaceHub  # Change back to HuggingFaceHub
 import datetime
 import os
 
@@ -44,12 +44,9 @@ def create_ai_news_crew():
         verbose=True,
         allow_delegation=False,
         tools=[search_tool],
-        llm=HuggingFaceEndpoint(
-            endpoint_url="https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-Coder-33B-Instruct",
-            task="text-generation",
-            temperature=0.5,  # Pass parameters explicitly
-            max_new_tokens=1024, # Pass parameters explicitly
-            repetition_penalty=1.2, # Pass parameters explicitly
+        llm=HuggingFaceHub(  # Use HuggingFaceHub
+            repo_id="deepseek-ai/DeepSeek-Coder-33B-Instruct", #Specify the model
+            model_kwargs={"temperature": 0.5, "max_new_tokens": 1024, "repetition_penalty": 1.2}, # Add model parameters here
             huggingfacehub_api_token=huggingfacehub_api_token,
         )
     )
@@ -63,12 +60,9 @@ def create_ai_news_crew():
             on AI developments quickly.""",
         verbose=True,
         allow_delegation=False,
-        llm=HuggingFaceEndpoint(
-            endpoint_url="https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-Coder-33B-Instruct",
-            task="text-generation",
-            temperature=0.2, # Pass parameters explicitly
-            max_new_tokens=1024, # Pass parameters explicitly
-            repetition_penalty=1.2, # Pass parameters explicitly
+        llm=HuggingFaceHub(  # Use HuggingFaceHub
+            repo_id="deepseek-ai/DeepSeek-Coder-33B-Instruct", #Specify model
+            model_kwargs={"temperature": 0.2, "max_new_tokens": 1024, "repetition_penalty": 1.2}, #Add model parameters here
             huggingfacehub_api_token=huggingfacehub_api_token,
         )
     )
@@ -130,7 +124,7 @@ def main():
 
             except Exception as e:
                 st.error(f"An error occurred: {e}")
-                st.error("Please check your API key, endpoint URL and ensure you have set up the environment correctly.")
+                st.error("Please check your API key, and ensure you have set up the environment correctly.")
 
 
 if __name__ == "__main__":