Update app.py

app.py

@@ -74,38 +74,24 @@ def infer(text):
         raise ValueError("Generated motion data is empty")
 
     gif_bytes = draw_to_batch([motion_data], [text], None)
-
-    # Ensure the directory exists in the current working directory
-    dir_path = "animation"
-    if not os.path.exists(dir_path):
-        os.makedirs(dir_path)
-
-    persistent_gif_path = os.path.join(dir_path, "animation.gif")
-    with open(persistent_gif_path, "wb") as f:
-        f.write(gif_bytes)
     
-    file_size = os.path.getsize(persistent_gif_path)
-    print(f"Saved GIF to {persistent_gif_path}, file size: {file_size} bytes")
-    if file_size <= 0:
-        raise ValueError("Generated GIF is empty")
-    if not os.path.exists(persistent_gif_path):
-        print(f"Error: GIF not saved to {persistent_gif_path}")
-    else:
-        print(f"GIF saved successfully to {persistent_gif_path}")
+    # Encode the GIF bytes to a base64 string
+    gif_base64 = base64.b64encode(gif_bytes).decode('ascii')
+    gif_html = f'<img src="data:image/gif;base64,{gif_base64}" />'
     
-    return persistent_gif_path
+    return gif_html
 
 with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") as demo:
     with gr.Column():
-        gr.Markdown("## 3D Human Motion" + ("Generation" if device == "cuda" else "CPU"))
+        gr.Markdown("## 3D Human Motion Generation")
         text_input = gr.Textbox(label="Enter the human action to generate", placeholder="Enter text description for the action here...")
-        output_image = gr.Image(label="Generated Human Motion", type='filepath')
+        output_html = gr.HTML(label="Generated Human Motion")
         submit_button = gr.Button("Generate Motion")
 
         submit_button.click(
             fn=infer,
             inputs=[text_input],
-            outputs=[output_image]
+            outputs=[output_html]
         )
 
 if __name__ == "__main__":