Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import torch
 import os
-from diffusers import StableDiffusion3Pipeline
+from diffusers import DiffusionPipeline
 from huggingface_hub import InferenceApi
 
 hf_api_key = os.getenv("PRODIGY_GA_02")
@@ -16,8 +16,8 @@ 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")
+model_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
+pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
 
 pipe.to("cuda")# If you have GPU access; otherwise, use "cpu"