Upload Example_demo.txt with huggingface_hub
Browse files- Example_demo.txt +1 -1
Example_demo.txt
CHANGED
|
@@ -14,7 +14,7 @@ model = model.half() # fix dtype mismatch
|
|
| 14 |
processor = AutoProcessor.from_pretrained("grohitraj/archive_classification")
|
| 15 |
|
| 16 |
# TEST
|
| 17 |
-
image = Image.open("
|
| 18 |
prompt = "<image>\nDescribe about the image for male aged 54:"
|
| 19 |
inputs = processor(text=prompt, images=image, return_tensors="pt")
|
| 20 |
inputs = {k: v.to(model.device) for k, v in inputs.items()}
|
|
|
|
| 14 |
processor = AutoProcessor.from_pretrained("grohitraj/archive_classification")
|
| 15 |
|
| 16 |
# TEST
|
| 17 |
+
image = Image.open("example_from_2019ISIC_data.jpg")
|
| 18 |
prompt = "<image>\nDescribe about the image for male aged 54:"
|
| 19 |
inputs = processor(text=prompt, images=image, return_tensors="pt")
|
| 20 |
inputs = {k: v.to(model.device) for k, v in inputs.items()}
|