Update app.py

app.py

@@ -1,9 +1,19 @@
-from google.colab import drive
 
-drive.mount("/content/drive")
 
 #Data Preprocessing
 
+import gradio as gr
+from tensorflow.keras.models import Model
+import matplotlib.pyplot as plt
+import cv2  # For color mapping the heatmap
+
+
+
+
+
+
+
+
 import os
 import numpy as np
 import tensorflow as tf
@@ -15,7 +25,7 @@ IMAGE_SIZE = (224, 224)
 BATCH_SIZE = 32
 
 # Paths to your dataset
-TRAIN_PATH = '/content/drive/MyDrive/archive/dataset'
+TRAIN_PATH = '/archive/dataset'
 
 # Data generator for loading and preprocessing images
 datagen = ImageDataGenerator(rescale=1./255, validation_split=0.15)
@@ -159,7 +169,7 @@ def preprocess_image(image_path):
     return img
 
 # Path to the image
-img_path = '/content/drive/MyDrive/archive/dataset/covid/01E392EE-69F9-4E33-BFCE-E5C968654078.jpeg'
+img_path = '/archive/dataset/covid/01E392EE-69F9-4E33-BFCE-E5C968654078.jpeg'
 
 # Preprocess the image
 img_array = preprocess_image(img_path)