Update app.py

app.py

@@ -1,3 +1,18 @@
 import gradio as gr
+from diffusers import DiffusionPipeline
 
-gr.load("models/stabilityai/stable-diffusion-2-1").launch()
+pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1")
+
+def predict(input_img):
+    predictions = pipeline(input_img)
+    return predictions
+
+gradio_app = gr.Interface(
+    predict,
+    inputs=gr.Text(label="Select hot dog candidate"),
+    outputs=[gr.Image(label="Processed Image"), gr.Label(label="Result", num_top_classes=2)],
+    title="Image Generator",
+)
+
+if __name__ == "__main__":
+    gradio_app.launch()