TedYeh commited on
Commit
f680b9e
1 Parent(s): a10bcbf

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from predictor import inference
3
 
4
- def index_predict(name):
5
- outputs, preds, heights, bust, waist, hips, description = inference(os.path.join(app.config['UPLOAD_FOLDER'], filename), epoch = 7)
6
  return heights, round(float(bust)), round(float(waist)), round(float(hips)), description[0], description[1]
7
 
8
  with gr.Blocks() as demo:
 
1
  import gradio as gr
2
  from predictor import inference
3
 
4
+ def index_predict(img):
5
+ outputs, preds, heights, bust, waist, hips, description = inference(img, epoch = 7)
6
  return heights, round(float(bust)), round(float(waist)), round(float(hips)), description[0], description[1]
7
 
8
  with gr.Blocks() as demo: