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:
|