Update app.py

app.py

@@ -1,24 +1,34 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusion3Pipeline
 import os
+from diffusers import StableDiffusion3Pipeline
 from huggingface_hub import InferenceApi
 
-
-# Retrieve Hugging Face API key from environment variables
 hf_api_key = os.getenv("PRODIGY_GA_02")
+if hf_api_key is None:
+    raise ValueError("Hugging Face API key 'PRODIGY_GA_02' not found. Ensure it is set as a secret.")
+
+# Initialize the Hugging Face API with the restricted model and token
+inference = InferenceApi(repo_id="stabilityai/stable-diffusion-3.5-medium", token=hf_api_key)
 
-# Load model with authentication token
+# Example inference request
+response = inference(inputs="Your input text here")
+print(response)
+
+# Load model
 model_id = "stabilityai/stable-diffusion-3.5-medium"
 pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium")
 
 pipe.to("cuda")# If you have GPU access; otherwise, use "cpu"
 
-# Define Gradio interface function
+
+
+# Define Gradio interface
 def generate_image(prompt):
     images = pipe(prompt).images
     return images[0]
 
+
 # Create Gradio UI
 iface = gr.Interface(
     fn=generate_image,
@@ -29,4 +39,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()