Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ def drop_incomplete_tail(text):
|
|
18 |
@spaces.GPU(duration=120)
|
19 |
def caption_image(image):
|
20 |
with torch.no_grad():
|
21 |
-
outputs = model([image],
|
22 |
return outputs
|
23 |
|
24 |
def greet(image):
|
|
|
18 |
@spaces.GPU(duration=120)
|
19 |
def caption_image(image):
|
20 |
with torch.no_grad():
|
21 |
+
outputs = model([image], 77).generated_text[0]
|
22 |
return outputs
|
23 |
|
24 |
def greet(image):
|