laurenssam commited on
Commit
d1f80dd
·
1 Parent(s): 47aeff6
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -105,7 +105,8 @@ def get_response(params):
105
 
106
  images = [load_image_from_base64(image) for image in images]
107
  print(images)
108
- images = process_images(images, image_processor, model.config)
 
109
 
110
  if type(images) is list:
111
  images = [
 
105
 
106
  images = [load_image_from_base64(image) for image in images]
107
  print(images)
108
+ # images = process_images(images, image_processor, model.config)
109
+ images_tensor = process_images(images, image_processor, model.config).to('cpu', dtype=torch.float16)
110
 
111
  if type(images) is list:
112
  images = [