Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import gradio.components as comp
 import os
 
 api_key = os.environ.get("HUGGINGFACE_API_KEY")
@@ -23,11 +24,11 @@ default_model = "stabilityai/stable-diffusion-2"
 model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
 
 def generate_image(text, model_name):
-    model = gr.interface.load_model(model_name, source="huggingface", api_key=api_key)
+    model = gr.Interface.load_model(model_name, source="huggingface", api_key=api_key)
     return model.predict(text)
 
 input_text = gr.inputs.Textbox(label="Input Text")
-output_image = gr.outputs.Image(label="Generated Image", type="pil")
+output_image = comp.Image(label="Generated Image")
 
 iface = gr.Interface(
     fn=generate_image,