Update app.py

app.py

@@ -98,17 +98,31 @@ def infer(filename):
 
     return result, plot_slices(2, 10, 128, 128, vol[0, :, :, :20])
 
-model = from_pretrained_keras('jalFaizy/3D_CNN')
-
-inputs = gr.inputs.File()
-outputs = [gr.outputs.Textbox(), 'plot']
-
-iface = gr.Interface(
-    infer, 
-    inputs,
-    outputs,
-    title='3D CNN for CT scans',
-    examples=['example_1_normal.nii.gz']
-)
-
-iface.launch()
+if __name__ == "__main__":
+    model = from_pretrained_keras('jalFaizy/3D_CNN')
+    
+    inputs = gr.inputs.File()
+    outputs = [gr.outputs.Textbox(), 'plot']
+    
+    title = "3D CNN for CT scans"
+    description = """This space implements 3D convolutional neural network to predict the presence of viral pneumonia in computer tomography scans.
+    """
+    article = """
+    <p style='text-align: center'>
+        <a href='https://keras.io/examples/vision/3D_image_classification/' target='_blank'>Keras Example given by Hasib Zunair</a>
+        <br>
+        Space by Faizan Shaikh
+    </p>
+    """
+    
+    iface = gr.Interface(
+        infer, 
+        inputs,
+        outputs,
+        title=title,
+        description=description,
+        article=article,
+        examples=['example_1_normal.nii.gz']
+    )
+    
+    iface.launch(enable_queue=True)