Update Chest_Xray_Report_Generator-V2.py

Chest_Xray_Report_Generator-V2.py

@@ -115,7 +115,7 @@ def reshape_transform(tensor, height=12, width=12):  ### height=14, width=14 for
 
 # Example usage:
 #image_path = "/home/chayan/CGI_Net/images/images/CXR1353_IM-0230-1001.png"
-model_path = "./Model/"
+model_path = "ChayanM/Swin-GPT2-Mimic-Impression"
 output_path = "./CAM-Result/"
 
 
@@ -187,7 +187,7 @@ def dicom_to_png(dicom_file, png_file):
     return img
 
 
-Image_Captioner = pipeline("image-to-text", model = "./Model/", device = 0)
+Image_Captioner = pipeline("image-to-text", model = "ChayanM/Swin-GPT2-Mimic-Impression", device = 0)
 
 data_dir = "./CAM-Result"