Update app.py

app.py

@@ -1,8 +1,14 @@
 import streamlit as st
-from transformers import pipeline
+import requests
 
-# Load the chatbot model using the "text-generation" task
-chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
+# Hugging Face Inference API endpoint
+API_URL = "https://api-inference.huggingface.co/models/microsoft/DialoGPT-medium"
+headers = {"Authorization": "Bearer YOUR_HUGGING_FACE_TOKEN"}
+
+def query_chatbot(prompt):
+    payload = {"inputs": prompt}
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.json()[0]['generated_text']
 
 # Streamlit app
 st.title("Electronics Component Chatbot")
@@ -25,7 +31,7 @@ if prompt := st.chat_input("What is your question?"):
         st.markdown(prompt)
 
     # Get chatbot response
-    response = chatbot(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
+    response = query_chatbot(prompt)
     
     # Add chatbot response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})