Show keypoints as text

app.py

@@ -2,7 +2,7 @@ import os
 os.system("pip install gradio==3.4")
 import gradio as gr
 os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git'")
-os.system("git clone https://github.com/ShuhongChen/bizarre-pose-estimator.git")
+os.system("git clone https://github.com/kivantium/bizarre-pose-estimator.git")
 os.chdir("bizarre-pose-estimator")
 
 os.system("wget https://i.imgur.com/IkJzlaE.jpeg")
@@ -16,14 +16,17 @@ os.system("cp -a ./bizarre_pose_models/. .")
 
 import urllib.request
 from urllib.parse import urlparse
+import numpy as np
 
 
 def inference(url):
   filename = os.path.basename(urlparse(url).path)
   urllib.request.urlretrieve(url, filename)
   os.system("python3 -m _scripts.pose_estimator "+filename+" ./_train/character_pose_estim/runs/feat_concat+data.ckpt")
+  keypoints = np.load("./_samples/keypoints.npy")
 
-  return "./_samples/character_pose_estim.png"
+
+  return ("./_samples/character_pose_estim.png", str(keypoints))
   
   
 title = "bizarre-pose-estimator"
@@ -34,7 +37,7 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.018
 gr.Interface(
     inference, 
     gr.Textbox(lines=1, placeholder="Image URL"),
-    gr.outputs.Image(type="file", label="Output"),
+    [gr.outputs.Image(type="file", label="Output"), gr.Textbox(lines=10, placeholder="Keypoints")],
     title=title,
     description=description,
     article=article,