Update app.py

app.py

@@ -14,11 +14,9 @@ import warnings
 import time
 import asyncio
 from functools import partial
-from concurrent.futures import ProcessPoolExecutor as Pool
 
 # external
 
-import spaces
 import torch
 import gradio as gr
 from numpy import asarray as array
@@ -64,10 +62,10 @@ result = []
 seq=512
 fast=True
 fps=20
-time=1
+time=3
 width=896
 height=896
-step=40
+step=50
 accu=8.5
 
 # ui data
@@ -177,7 +175,6 @@ def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU(duration=180)
 def pipe_generate(img,p1,p2,motion):
     global last_motion
     global pipe
@@ -201,7 +198,7 @@ def pipe_generate(img,p1,p2,motion):
             guidance_scale=accu,
             num_inference_steps=step,
             max_sequence_length=seq,
-            generator=torch.Generator("cuda").manual_seed(0)
+            generator=torch.Generator(device).manual_seed(0)
         ).images[0]
 
     return pipe(
@@ -251,15 +248,15 @@ def ui():
         with gr.Row(elem_id="col-container"):
             with gr.Column():
                 with gr.Row():
-                    img = gr.Image(label="STATIC PHOTO",show_label=True,container=True,type="pil")
+                    img = gr.Image(label="UPLOAD PHOTO",show_label=True,container=True,type="pil")
                 with gr.Row():
                     prompt = gr.Textbox(
                         elem_id="prompt",
-                        placeholder="PROMPT",
+                        placeholder="INCLUDE",
                         container=False,
                         max_lines=1
                     )
-                with gr.Row(visible=False):
+                with gr.Row():
                     prompt2 = gr.Textbox(
                         elem_id="prompt2",
                         placeholder="EXCLUDE",