Update app.py

app.py

@@ -2,9 +2,7 @@ import gradio as gr
 from langchain.prompts import ChatPromptTemplate
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain.schema import HumanMessage, SystemMessage
-from langchain.callbacks.base import BaseCallbackHandler
-import os
-import time
+from langchain.callbacks
 
 # Custom streaming callback for Gradio
 class GradioStreamingCallback(BaseCallbackHandler):
@@ -38,11 +36,8 @@ llm = HuggingFaceEndpoint(
     repo_id="HuggingFaceH4/zephyr-7b-beta",
     temperature=0.7,
     top_p=0.9,
-    model_kwargs={
-        "max_length": 1536,
-        "return_full_text": False,
-        "repetition_penalty": 1.1,
-    },
+    repetition_penalty=1.1,
+    model_kwargs={"max_length": 1536},
     huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
 )