Update app.py

app.py

@@ -3,9 +3,11 @@ import streamlit as st
 from phi.assistant import Assistant
 from phi.tools.arxiv_toolkit import ArxivToolkit
 from huggingface_hub import InferenceClient
+from phi.llm.huggingface import HuggingFaceLLM  # Added for proper LLM wrapping
 
 # Initialize the Hugging Face Inference Client
-client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta")
+raw_client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta")
+client = HuggingFaceLLM(client=raw_client)  # Wrap the client properly
 
 # Set up the Streamlit app
 st.set_page_config(page_title="Chat with Research Papers", layout="wide")