Huuvang commited on
Commit
0d0ea38
·
verified ·
1 Parent(s): 5e7d6b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -111,7 +111,7 @@ def generate_response(image, question):
111
  if isinstance(image, str):
112
  image = Image.open(image).convert('RGB')
113
  else:
114
- image = Image.fromarray(image).convert('RGB')
115
 
116
  # Process image
117
  pixel_values = load_image(image, max_num=6).to(torch.bfloat16).to(device)
 
111
  if isinstance(image, str):
112
  image = Image.open(image).convert('RGB')
113
  else:
114
+ image = image.convert("RGB")
115
 
116
  # Process image
117
  pixel_values = load_image(image, max_num=6).to(torch.bfloat16).to(device)