Update app.py

app.py

@@ -5,7 +5,8 @@ from langchain.llms import HuggingFaceHub
 def generate_answer(query):
     llm = HuggingFaceHub(
         repo_id="TheBloke/goliath-120b-GGUF",
-        model_kwargs={"temperature": 0.5, "max_length": 64, "max_new_tokens": 512}
+        model_kwargs={"temperature": 0.5, "max_length": 64, "max_new_tokens": 512},
+        task="text2text-generation"
     )
     prompt = f"""
         You are a helpful AI assistant.