Spaces:
Runtime error
Runtime error
artelabsuper
commited on
Commit
·
a05fab5
1
Parent(s):
f5085fa
fix for timing reason
Browse files
app.py
CHANGED
|
@@ -47,14 +47,14 @@ def predict(input_image):
|
|
| 47 |
data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
|
| 48 |
data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
|
| 49 |
|
| 50 |
-
return data
|
| 51 |
|
| 52 |
iface = gr.Interface(
|
| 53 |
fn=predict,
|
| 54 |
inputs=gr.Image(shape=(512,512)),
|
| 55 |
outputs=[
|
| 56 |
gr.Image(shape=(512,512)),
|
| 57 |
-
gr.outputs.Textbox(label='Raw output')
|
| 58 |
],
|
| 59 |
examples=[
|
| 60 |
["demo_imgs/fake.jpg"] # use real image
|
|
|
|
| 47 |
data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
|
| 48 |
data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
|
| 49 |
|
| 50 |
+
return data #, str(predicted_image.tolist())
|
| 51 |
|
| 52 |
iface = gr.Interface(
|
| 53 |
fn=predict,
|
| 54 |
inputs=gr.Image(shape=(512,512)),
|
| 55 |
outputs=[
|
| 56 |
gr.Image(shape=(512,512)),
|
| 57 |
+
# gr.outputs.Textbox(label='Raw output')
|
| 58 |
],
|
| 59 |
examples=[
|
| 60 |
["demo_imgs/fake.jpg"] # use real image
|