Spaces:

yutengz
/

Action2Vision

Runtime error

yutengz commited on May 5, 2025

Commit

3f19484

verified ·

1 Parent(s): e7a7448

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
 # Load model from HF Hub (your model repo)
-model_id = "yutengz/ip2p-RoboPredict"
 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
     model_id,
@@ -26,7 +26,7 @@ demo = gr.Interface(
         gr.Textbox(label="Instruction Prompt", placeholder="e.g., stack the blocks"),
     ],
     outputs=gr.Image(label="Predicted Image"),
-    title="🧠 ip2p-RoboPredict",
     description="A fine-tuned InstructPix2Pix model for robotic action frame prediction."
 )

 from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
 # Load model from HF Hub (your model repo)
+model_id = "yutengz/Action2Vision"
 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
     model_id,
         gr.Textbox(label="Instruction Prompt", placeholder="e.g., stack the blocks"),
     ],
     outputs=gr.Image(label="Predicted Image"),
+    title="🧠 Action2Vision",
     description="A fine-tuned InstructPix2Pix model for robotic action frame prediction."
 )