Update app.py

app.py

@@ -1,6 +1,18 @@
 import gradio as gr
 import torch
 from diffusers import StableDiffusion3Pipeline
+from huggingface_hub import InferenceApi
+
+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)
+
+# Example inference request
+response = inference(inputs="Your input text here")
+print(response)
 
 # Load model
 model_id = "stabilityai/stable-diffusion-3.5-medium"