Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,8 +26,8 @@ model = model.to(device)
|
|
26 |
# Deploy
|
27 |
title = 'Image Captioning'
|
28 |
description = 'Huỳnh Công Chánh'
|
29 |
-
inputs
|
30 |
-
outputs
|
31 |
|
32 |
def inference(raw_image):
|
33 |
image = transform(raw_image).unsqueeze(0).to(device)
|
|
|
26 |
# Deploy
|
27 |
title = 'Image Captioning'
|
28 |
description = 'Huỳnh Công Chánh'
|
29 |
+
inputs = gr.Image(type='pil')
|
30 |
+
outputs = gr.Textbox(label='Output')
|
31 |
|
32 |
def inference(raw_image):
|
33 |
image = transform(raw_image).unsqueeze(0).to(device)
|