Update app.py

app.py

@@ -1,11 +1,12 @@
+
 import os
 import gradio as gr
-from langchain_community.llms import HuggingFaceHub
+from langchain_community.llms import HuggingFaceEndpoint
 from langchain.prompts import PromptTemplate
 
 # Initialize the chatbot
 HF_TOKEN = os.getenv("HF_TOKEN")
-llm = HuggingFaceHub(
+llm = HuggingFaceEndpoint(
     repo_id="google/gemma-1.1-7b-it",
     task="text-generation",
     model_kwargs={
@@ -25,7 +26,7 @@ QA_CHAIN_PROMPT = PromptTemplate(input_variables=["context", "question"],templat
 
 def predict(message, history):
     input_prompt = QA_CHAIN_PROMPT.format(question=message, context=history)
-    ai_msg = llm.generate(input_prompt)
+    ai_msg = llm.generate([input_prompt])  # Pass a list of strings
     return ai_msg
 
 gr.ChatInterface(predict).launch()