Update app.py

app.py

@@ -1,26 +1,3 @@
 import gradio as gr
-import base64
-from io import BytesIO
-from PIL import Image
 
-# Function to convert image to base64
-def image_to_base64(image: Image.Image):
-    buffered = BytesIO()
-    image.save(buffered, format="PNG")  # You can change the format if needed
-    return base64.b64encode(buffered.getvalue()).decode('utf-8')
-
-# Function to generate the image and return base64-encoded response
-def generate_base64_image(prompt: str):
-    # Load the pre-trained model (adjust to your needs)
-    model = gr.load("models/stabilityai/stable-diffusion-3.5-large")
-    
-    # Generate image with the model
-    img = model.predict(prompt)[0]  # Assuming the model returns the generated image
-    
-    # Convert image to base64
-    return image_to_base64(img)
-
-# Define Gradio interface
-iface = gr.Interface(fn=generate_base64_image, inputs="text", outputs="text")  # Outputs base64 text
-
-iface.launch()
+gr.load("models/stabilityai/stable-diffusion-3.5-large").launch()