Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,8 +9,6 @@ def generate(textI, texti1, texti2):
|
|
9 |
plot_3d_motion("result.mp4", HML_KINEMATIC_CHAIN, preds, title="", fps=30)
|
10 |
return "result.mp4"
|
11 |
|
12 |
-
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
13 |
-
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
14 |
model = AutoModel.from_pretrained("pabloruizponce/in2IN", trust_remote_code=True)
|
15 |
model.to("cuda")
|
16 |
demo = gr.Interface(fn=generate,
|
@@ -18,4 +16,5 @@ demo = gr.Interface(fn=generate,
|
|
18 |
gr.Text(label="Individual1 Description"),
|
19 |
gr.Text(label="Individual2 Description")],
|
20 |
outputs=gr.Video())
|
|
|
21 |
demo.launch()
|
|
|
9 |
plot_3d_motion("result.mp4", HML_KINEMATIC_CHAIN, preds, title="", fps=30)
|
10 |
return "result.mp4"
|
11 |
|
|
|
|
|
12 |
model = AutoModel.from_pretrained("pabloruizponce/in2IN", trust_remote_code=True)
|
13 |
model.to("cuda")
|
14 |
demo = gr.Interface(fn=generate,
|
|
|
16 |
gr.Text(label="Individual1 Description"),
|
17 |
gr.Text(label="Individual2 Description")],
|
18 |
outputs=gr.Video())
|
19 |
+
|
20 |
demo.launch()
|