Staticaliza commited on
Commit
6c15fcf
·
verified ·
1 Parent(s): 3538bec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -32,10 +32,10 @@ def generate(image, instruction=DEFAULT_INPUT, sampling=False, temperature=0.7,
32
  repo.to(DEVICE)
33
 
34
  print(image)
35
- # image_rgb = Image.open(image).convert("RGB")
36
- # print(image_rgb, instruction)
37
 
38
- inputs = [{"role": "user", "content": [image, instruction]}]
39
 
40
  parameters = {
41
  "sampling": sampling,
 
32
  repo.to(DEVICE)
33
 
34
  print(image)
35
+ image_data = Image.fromarray(image_array.astype('uint8'), 'RGB')
36
+ print(image_data, instruction)
37
 
38
+ inputs = [{"role": "user", "content": [image_data, instruction]}]
39
 
40
  parameters = {
41
  "sampling": sampling,