Linoy Tsaban commited on
Commit
0e70f76
1 Parent(s): c9dd9ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image
26
  ## IMAGE CPATIONING ##
27
  def caption_image(input_image):
28
 
29
- inputs = blip_processor(images=input_image, return_tensors="pt")
30
  pixel_values = inputs.pixel_values
31
 
32
  generated_ids = blip_model.generate(pixel_values=pixel_values, max_length=50)
 
26
  ## IMAGE CPATIONING ##
27
  def caption_image(input_image):
28
 
29
+ inputs = blip_processor(images=input_image, return_tensors="pt").to(device)
30
  pixel_values = inputs.pixel_values
31
 
32
  generated_ids = blip_model.generate(pixel_values=pixel_values, max_length=50)