Update Chest_Xray_Report_Generator-V2.py

Chest_Xray_Report_Generator-V2.py

@@ -37,10 +37,10 @@ from openai import OpenAI
 # client = OpenAI()
 client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
 
-import spaces  # Import the spaces module for ZeroGPU
+# import spaces  # Import the spaces module for ZeroGPU
 
 
-@spaces.GPU 
+# @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,
@@ -191,7 +191,7 @@ Image_Captioner = pipeline("image-to-text", model = "./Model/", device = 0)
 
 data_dir = "./CAM-Result"
 
-@spaces.GPU(duration=300)
+# @spaces.GPU(duration=300)
 def xray_report_generator(Image_file, Query):
   if Image_file[-4:] =='.dcm':
     png_file = 'DCM2PNG.png'
@@ -230,7 +230,7 @@ def xray_report_generator(Image_file, Query):
 
 
 def save_feedback(feedback):
-    feedback_dir = "Chayan/Feedback/"  # Update this to your desired directory
+    feedback_dir = "./Feedback/"  # Update this to your desired directory
     if not os.path.exists(feedback_dir):
         os.makedirs(feedback_dir)
     feedback_file = os.path.join(feedback_dir, "feedback.txt")