keithhon commited on
Commit
e7e1448
·
1 Parent(s): 3943d19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def image_feature(url):
10
  image = Image.open(requests.get(url, stream=True).raw)
11
  inputs = processor(images=image, return_tensors="pt")
12
  image_features = model.get_image_features(**inputs)
13
- return image_features.numpy().tolist()[0]
14
 
15
  iface = gr.Interface(fn=image_feature, inputs="text", outputs="text")
16
  iface.launch()
 
10
  image = Image.open(requests.get(url, stream=True).raw)
11
  inputs = processor(images=image, return_tensors="pt")
12
  image_features = model.get_image_features(**inputs)
13
+ return image_features.detach().numpy().tolist()[0]
14
 
15
  iface = gr.Interface(fn=image_feature, inputs="text", outputs="text")
16
  iface.launch()