Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 import os
 # Load Stable Diffusion model
 HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None
-model = gr.load("models/stabilityai/stable-diffusion-3.5-large", hf_token=HF_TOKEN)
+model = gr.Interface.load("models/stabilityai/stable-diffusion-3.5-large", hf_token=HF_TOKEN)
 
 # Chevereto API details
 CHEVERETO_API_URL = os.environ.get("api_url")  # Replace with your Chevereto API endpoint
@@ -35,7 +35,7 @@ def upload_to_chevereto(image: Image.Image) -> str:
 
 # Generate and upload the image
 def generate_image_and_upload(prompt: str) -> str:
-    img = model.predict(prompt)[0]  # Get generated image
+    img = img = model.submit([prompt])  # Get generated image
     return upload_to_chevereto(img)  # Upload and return the URL
 
 # Define Gradio API returning the Chevereto public URL