Update app.py

app.py

@@ -21,7 +21,7 @@ conditions = [
 ]
 
 # Load and configure the model
-model = models.densenet121(pretrained=True)  # DenseNet pre-trained on ImageNet
+model = models.densenet121(weights="IMAGENET1K_V1")  # DenseNet pre-trained on ImageNet
 num_features = model.classifier.in_features
 model.classifier = torch.nn.Linear(num_features, len(conditions))  # Output for all 24 conditions
 model.eval()
@@ -158,6 +158,7 @@ def create_interface():
         gr.Markdown("<p style='text-align: center; color: #666;'>AI-powered analysis for X-rays and patient reports</p>")
         
         with gr.TabbedInterface() as tabs:
+            # Tab for X-ray Analysis
             with gr.Tab("X-ray Analysis"):
                 with gr.Row():
                     xray_input = gr.Image(label="Upload X-ray", type="pil", elem_id="xray-input")
@@ -182,6 +183,7 @@ def create_interface():
                     outputs=[xray_input]
                 )
             
+            # Tab for Report Analysis
             with gr.Tab("Report Analysis"):
                 with gr.Row():
                     report_input = gr.File(label="Upload Patient Report (PDF)", file_count="single", elem_id="report-input")