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 HuggingFaceHub
+from langchain_community.llms import HuggingFaceEndpoint  # Changed to HuggingFaceEndpoint
 import datetime
 import os
 
@@ -44,9 +44,9 @@ def create_ai_news_crew():
         verbose=True,
         allow_delegation=False,
         tools=[search_tool],
-        llm=HuggingFaceHub(
-            repo_id="deepseek-ai/DeepSeek-Coder-33B-Instruct",
-            task="text-generation", # Explicitly set the task
+        llm=HuggingFaceEndpoint(  # Use HuggingFaceEndpoint
+            endpoint_url="https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-Coder-33B-Instruct", #Use correct end point
+            task="text-generation",
             model_kwargs={"temperature": 0.5, "max_new_tokens": 1024, "repetition_penalty": 1.2},
             huggingfacehub_api_token=huggingfacehub_api_token,
         )
@@ -61,9 +61,9 @@ def create_ai_news_crew():
             on AI developments quickly.""",
         verbose=True,
         allow_delegation=False,
-        llm=HuggingFaceHub(
-            repo_id="deepseek-ai/DeepSeek-Coder-33B-Instruct",
-            task="text-generation", # Explicitly set the task
+        llm=HuggingFaceEndpoint(  # Use HuggingFaceEndpoint
+            endpoint_url="https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-Coder-33B-Instruct", #Use correct endpoint
+            task="text-generation",
             model_kwargs={"temperature": 0.2, "max_new_tokens": 1024, "repetition_penalty": 1.2},
             huggingfacehub_api_token=huggingfacehub_api_token,
         )
@@ -126,7 +126,7 @@ def main():
 
             except Exception as e:
                 st.error(f"An error occurred: {e}")
-                st.error("Please check your API key and ensure you have set up the environment correctly.")
+                st.error("Please check your API key, endpoint URL and ensure you have set up the environment correctly.") #Added endpoint url check
 
 
 if __name__ == "__main__":