Update Chest_Xray_Report_Generator-V2.py
Browse files
Chest_Xray_Report_Generator-V2.py
CHANGED
|
@@ -125,7 +125,7 @@ def dicom_to_png(dicom_file, png_file):
|
|
| 125 |
return img
|
| 126 |
|
| 127 |
|
| 128 |
-
Image_Captioner = pipeline("image-to-text", model = "./Mimic_test/")
|
| 129 |
|
| 130 |
data_dir = "./CAM-Result"
|
| 131 |
|
|
|
|
| 125 |
return img
|
| 126 |
|
| 127 |
|
| 128 |
+
Image_Captioner = pipeline("image-to-text", model = "./Mimic_test/", device = 0)
|
| 129 |
|
| 130 |
data_dir = "./CAM-Result"
|
| 131 |
|