Files

app.py

@@ -1,12 +1,5 @@
-import os
-import subprocess
-import shlex
-from src.demo.model import DesignEdit
 import spaces
 
-from src.demo.demo import *
-import shlex
-
 import cv2
 import gradio as gr
 
@@ -19,8 +12,6 @@ pipe.to('cuda')
 def generate(prompt):
     return pipe(prompt).images
 
-gr.Image()
-
 gr.Interface(
     fn=generate,
     inputs=gr.Text(),