Update app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 import torch
 from diffusers import StableDiffusion3Pipeline
 import os
+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.")
 
 # Load model with authentication token
 model_id = "stabilityai/stable-diffusion-3.5-medium"