kivantium commited on
Commit
273d51b
·
unverified ·
1 Parent(s): 986cced

Change input

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -16,10 +16,12 @@ os.system("cp -a ./bizarre_pose_models/. .")
16
 
17
  os.system("ls")
18
 
19
- def inference(img):
20
- os.system("python3 -m _scripts.pose_estimator "+img+" ./_train/character_pose_estim/runs/feat_concat+data.ckpt")
21
 
22
- return "./_samples/character_pose_estim.png"
 
 
23
 
24
 
25
  title = "bizarre-pose-estimator"
@@ -30,8 +32,8 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.018
30
  examples=[["IkJzlaE.jpeg"]]
31
  gr.Interface(
32
  inference,
33
- gr.inputs.Image(type="filepath", label="Input"),
34
- gr.outputs.Image(type="file", label="Output"),
35
  title=title,
36
  description=description,
37
  article=article,
 
16
 
17
  os.system("ls")
18
 
19
+ def inference(urls):
20
+ #os.system("python3 -m _scripts.pose_estimator "+img+" ./_train/character_pose_estim/runs/feat_concat+data.ckpt")
21
 
22
+ #return "./_samples/character_pose_estim.png"
23
+
24
+ return urls
25
 
26
 
27
  title = "bizarre-pose-estimator"
 
32
  examples=[["IkJzlaE.jpeg"]]
33
  gr.Interface(
34
  inference,
35
+ inputs=gr.Textbox(lines=5, placeholder="Image URL"),
36
+ outputs=gr.Textbox(lines=5, placeholder="Result"),
37
  title=title,
38
  description=description,
39
  article=article,