Spaces:
Running
Running
TedYeh
commited on
Commit
•
f680b9e
1
Parent(s):
a10bcbf
update app
Browse files
app.py
CHANGED
@@ -1,8 +1,8 @@
|
|
1 |
import gradio as gr
|
2 |
from predictor import inference
|
3 |
|
4 |
-
def index_predict(
|
5 |
-
outputs, preds, heights, bust, waist, hips, description = inference(
|
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:
|