Update app.py

app.py

@@ -1,11 +1,12 @@
 import os
 import streamlit as st
-from huggingface_hub import InferenceApi, login
+from huggingface_hub import InferenceApi, login, InferenceClient
+
 
 # Streamlit app configuration
 st.set_page_config(page_title="Medical Chatbot")
-st.title("Medical Chatbot")
-st.subheader("I'm Your Medical Assistance", divider='rainbow')
+st.title("Medical Chatbot 🤖")
+st.subheader("", divider='rainbow')
 # Get the Hugging Face token from environment variables
 hf_token = os.getenv("HF_TOKEN")
 if hf_token is None:
@@ -32,7 +33,7 @@ st.sidebar.markdown(model_info[selected_model]['description'])
 st.sidebar.image(model_info[selected_model]['logo'])
 
 # Inference API Initialization
-client = InferenceApi(repo_id=model_links[selected_model], token=hf_token)
+client = InferenceClient('HuggingFaceH4/zephyr-7b-beta')
 
 # Sidebar settings
 max_tokens = st.sidebar.slider("Max new tokens", 1, 2048, 512)