Update Chest_Xray_Report_Generator-V2.py

Chest_Xray_Report_Generator-V2.py

@@ -7,6 +7,8 @@ import numpy as np
 import pydicom
 import time
 
+import spaces  # Import the spaces module for ZeroGPU
+
 ##### Libraries For Grad-Cam-View
 import os
 import cv2
@@ -19,6 +21,7 @@ from pytorch_grad_cam.utils.image import show_cam_on_image, preprocess_image
 from pytorch_grad_cam.ablation_layer import AblationLayerVit
 from transformers import VisionEncoderDecoderModel
 
+@spaces.GPU 
 def generate_gradcam(image_path, model_path, output_path, method='gradcam', use_cuda=True, aug_smooth=False, eigen_smooth=False):
     methods = {
         "gradcam": GradCAM,
@@ -126,6 +129,7 @@ Image_Captioner = pipeline("image-to-text", model = "./Mimic_test/")
 
 data_dir = "./CAM-Result"
 
+@spaces.GPU(duration=300)
 def xray_report_generator(Image_file):
   if Image_file[-4:] =='.dcm':
     png_file = 'DCM2PNG.png'