Update app.py

app.py

@@ -236,7 +236,8 @@ def sample(prompt):
     image = get_images(up_samples)
     # image = to_base64(image)
     # return {"image": image}
-    return image
+    resized_image = image.resize((128, 128), Image.ANTIALIAS)
+    return resized_image
 
 
 
@@ -248,7 +249,7 @@ examples =["melanoma"]
 
 iface = gr.Interface(fn=sample, 
                      inputs=gr.inputs.Textbox(label='Which dermoscopic entity would you like to see? Choose one of the following one: "melanoma", "melanocytic nevi", "Actinic keratoses and intraepithelial carcinoma / Bowen disease, "benign keratosis-like lesions (solar lentigines / seborrheic keratoses and lichen-planus like keratoses", "basal cell carcinoma", "dermatofibroma", "vascular lesions"'), 
-                     outputs=gr.outputs.Image(type="pil", label="Model input + completions").style(height=256, width=256),
+                     outputs=gr.outputs.Image(type="pil", label="Model input + completions").style(height=128, width=128),
                      title=title,
                      description=description,
                      article=article,