Update app.py

app.py

@@ -19,7 +19,7 @@ install_packages()
 
 import os
 import streamlit as st
-from langchain.llms import HuggingFaceHub
+from langchain_community.llms import HuggingFaceHub
 from langchain.schema import SystemMessage, HumanMessage, AIMessage
 import dotenv
 
@@ -34,7 +34,7 @@ if not token:
 
 llm = HuggingFaceHub(
     repo_id=repo_id,
-    model_kwargs={"max_length": 128, "temperature": 0.8},  # Adjust the temperature for response style
+    model_kwargs={"max_length": 512, "temperature": 0.8},  # Adjust the temperature for response style
     huggingfacehub_api_token=token,
 )
 
@@ -57,8 +57,11 @@ def init_messages() -> None:
         ]
 
 def get_answer(llm, user_input: str) -> str:
-    """Generates an answer from the model."""
-    return llm(user_input)
+    try:
+        return llm(user_input)
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
 
 def main() -> None:
     """Main function for the Streamlit app."""