Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline
-from huggingface_hub import hf_api
+import os
 
 # Retrieve the Hugging Face token stored in Hugging Face Spaces secrets
-HUGGINGFACE_TOKEN = hf_api.get_secret("keyss")
+HUGGINGFACE_TOKEN = os.getenv("keyss")
 
 if not HUGGINGFACE_TOKEN:
-    raise ValueError("Hugging Face token not found! Make sure it's added in the Hugging Face Secrets.")
+    raise ValueError("Hugging Face token not found! Make sure it's added in Hugging Face Secrets.")
 
 def image_generation(prompt):
     # Check if GPU is available