pabloruizponce commited on
Commit
4a95f54
·
verified ·
1 Parent(s): e7b9dae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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()