Update app.py

app.py

@@ -1,11 +1,5 @@
 import streamlit as st
 from transformers import pipeline
-import os
-from huggingface_hub import login
-
-# Hugging Face API Key
-api_key = "your_hugging_face_api_key"  # Replace with your API key
-login(api_key)
 
 # Title of the application
 st.title("AI Code Bot")
@@ -23,7 +17,11 @@ user_input = st.text_area("Enter your prompt (e.g., 'Write a Python function to
 @st.cache_resource
 def load_model():
     # Use a Hugging Face text-generation model
-    model = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", use_auth_token=api_key)
+    model = pipeline(
+        "text-generation",
+        model="EleutherAI/gpt-neo-1.3B",
+        use_auth_token=True  # Automatically uses Hugging Face Space's token
+    )
     return model
 
 model = load_model()