Spaces:

lyttt
/

VLV_Caption

Sleeping

lyttt commited on Jun 7

Commit

8132819

1 Parent(s): 841a2e8

.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ def greet(image):
     image = Image.fromarray(image, mode='RGB')
     with torch.no_grad():
         outputs = model([image]).generated_text[0]
-    return output
 demo = gr.Interface(fn=greet, inputs="image", outputs="text")
 demo.launch()

     image = Image.fromarray(image, mode='RGB')
     with torch.no_grad():
         outputs = model([image]).generated_text[0]
+    return outputs
 demo = gr.Interface(fn=greet, inputs="image", outputs="text")
 demo.launch()