Update app.py

app.py

@@ -11,8 +11,8 @@ st.title("ChatGPT-like Chatbot")
 base_url="https://api-inference.huggingface.co/models/"
 
 API_KEY = os.environ.get('HUGGINGFACE_API_KEY')
-print(API_KEY)
-headers = {"Authorization":"Bearer "+API_KEY}
+# print(API_KEY)
+# headers = {"Authorization":"Bearer "+API_KEY}
 
 model_links ={
     "Mistral-7B":base_url+"mistralai/Mistral-7B-Instruct-v0.2",
@@ -134,8 +134,8 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
         client = InferenceClient(
-            model=model_links[selected_model],
-            headers=headers)
+            model=model_links[selected_model],)
+            # headers=headers)
 
         output = client.text_generation(
             formated_text,