[CLEANUP]

servers/text_to_video/test.py

@@ -5,7 +5,7 @@
     EulerDiscreteScheduler,
     MotionAdapter,
 )
-from diffusers.utils import export_to_gif, export_to_video
+from diffusers.utils import export_to_gif
 from dotenv import load_dotenv
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
@@ -68,10 +68,9 @@ def text_to_video(
                 num_inference_steps=inference_steps,
             )
             outputs.append(output)
-            if output_type == ".gif":
-                out = export_to_gif([output], f"{output_path}_{i}.gif")
-            else:
-                out = export_to_video([output], f"{output_path}_{i}.mp4")
+            out = export_to_gif([output], f"{output_path}_{i}.gif")
+            # else:
+            #     out = export_to_video([output], f"{output_path}_{i}.mp4")
         # output = pipe(
         #     prompt = task,
         #     guidance_scale = guidance_scale,
@@ -84,5 +83,5 @@ def text_to_video(
         return None
 
 
-out = text_to_video(task="A girl in hijab studying in a library", n=5)
+out = text_to_video(task="A girl in hijab studying in a library")
 print(out)