Update app.py

app.py

@@ -23,13 +23,12 @@ import jax
 import jax.numpy as jnp
 
 last_motion=None
-fps=20
-time=16
+fps=10
+time=2
 device = "cuda"
 dtype = torch.float16
 result=[]
 step = 2
-progress=gr.Progress()
 
 css="""
 input, input::placeholder {
@@ -72,9 +71,9 @@ function custom(){
 }
 """
 
-def forest_schnell():
-    PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
-    return PIPE
+#def forest_schnell():
+#    PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
+#    return PIPE
     
 def translate(text,lang):
     if text == None or lang == None:
@@ -117,10 +116,6 @@ def translate(text,lang):
     print(ret)
     return ret
 
-def progress_callback(i, t, z):
-    global progress
-    progress((i+1, step))
-
 def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
@@ -131,7 +126,6 @@ def Piper(name,positive_prompt,negative,motion):
     global fps
     global time
     global last_motion
-    global progress
     global base
     global device
 
@@ -142,7 +136,7 @@ def Piper(name,positive_prompt,negative,motion):
             pipe.set_adapters(["motion"], [0.7])
         last_motion = motion
 
-    out = pipe(
+    return pipe(
         positive_prompt,
         negative_prompt=negative,
         height=1024,
@@ -152,9 +146,6 @@ def Piper(name,positive_prompt,negative,motion):
         num_frames=(fps*time)
     )
 
-    export_to_gif(out.frames[0],name,fps=fps)
-    return name
-
 def infer(pm):
         print("infer: started")
     
@@ -167,7 +158,9 @@ def infer(pm):
             _do.append(f'{p1}')
         posi = " ".join(_do)+neg
 
-        return Piper(name,posi,neg,pm["m"])
+        out = Piper(name,posi,neg,pm["m"])
+        export_to_gif(out.frames[0],name,fps=fps)
+        return name
 
 def run(m,p1,p2,*result):