Update app.py

app.py

@@ -1,16 +1,17 @@
 from transformers import pipeline
-from huggingface_hub import login
 
-# Login to Hugging Face with your API key
-api_key = 'your-huggingface-api-key'
-login(token=api_key)
+# Define your API key
+api_key = "your_huggingface_api_key"
 
-# Initialize the pipeline with a correct model identifier
-# Here I'm using a popular image generation model like 'stable-diffusion'
-generator = pipeline('image-generation', model='CompVis/stable-diffusion-v1-4-original', use_auth_token=api_key)
+# Initialize the pipeline using the API key
+generator = pipeline(
+    "image-generation",
+    model="CompVis/stable-diffusion-v1-4-original",
+    use_auth_token=api_key
+)
 
-# Generate an image from a text prompt
-generated_image = generator("A futuristic city")
+# Generate an image
+result = generator("A futuristic city")
 
-# Show the generated image
-generated_image[0]['image'].show()
+# Display the generated image
+result[0]["image"].show()