Update app.py

app.py

@@ -1,7 +1,5 @@
 import torch
 from torch import nn
-from huggingface_hub import hf_hub_download
-from torchvision.utils import save_image
 import gradio as gr
 
 
@@ -41,6 +39,6 @@ gr.Interface(
         gr.Slider(0, 3, label='Top', step=1, default=0),
         gr.Slider(0, 4, label='Bottom', step=1, default=0),
     ],
-    outputs="name",
+    outputs="image",
     live=True,
 ).launch(cache_examples=True)