Update app.py

app.py

@@ -1,22 +1,10 @@
-import torch
-from diffusers import DiffusionPipeline
-from diffusers.utils import load_image, export_to_video
-
-# pipe = DiffusionPipeline.from_pretrained("Wan-AI/Wan2.2-I2V-A14B", torch_dtype=torch.float16)
-# pipe.to("cuda")
-# pipe = DiffusionPipeline.from_pretrained("Wan-AI/Wan2.2-I2V-A14B")
-# pipe.to("cpu")
-
-pipe = DiffusionPipeline.from_pretrained("Linaqruf/animov-512x", torch_dtype=torch.float32)
-pipe.to("cpu")
-
-
-prompt = "A man with short gray hair plays a red electric guitar."
-image = load_image(
-    "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"
-)
-
-
-
-output = pipe(image=image, prompt=prompt).frames[0]
-export_to_video(output, "output.mp4")
+import gradio as gr
+
+with gr.Blocks(fill_height=True) as demo:
+    with gr.Sidebar():
+        gr.Markdown("# Inference Provider")
+        gr.Markdown("This Space showcases the Wan-AI/Wan2.2-I2V-A14B model, served by the fal-ai API. Sign in with your Hugging Face account to use this API.")
+        button = gr.LoginButton("Sign in")
+    gr.load("models/Wan-AI/Wan2.2-I2V-A14B", accept_token=button, provider="fal-ai")
+    
+demo.launch()