Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def generate(image, instruction=DEFAULT_INPUT, sampling=False, temperature=0.7,
|
|
32 |
repo.to(DEVICE)
|
33 |
|
34 |
print(image)
|
35 |
-
image_data = Image.fromarray(
|
36 |
print(image_data, instruction)
|
37 |
|
38 |
inputs = [{"role": "user", "content": [image_data, instruction]}]
|
|
|
32 |
repo.to(DEVICE)
|
33 |
|
34 |
print(image)
|
35 |
+
image_data = Image.fromarray(image.astype('uint8'), 'RGB')
|
36 |
print(image_data, instruction)
|
37 |
|
38 |
inputs = [{"role": "user", "content": [image_data, instruction]}]
|