Update pages/Chest.py

pages/Chest.py

@@ -19,7 +19,8 @@ from transformers import BioGptTokenizer, BioGptForCausalLM
 text_model = BioGptForCausalLM.from_pretrained("microsoft/biogpt")
 tokenizer = BioGptTokenizer.from_pretrained("microsoft/biogpt")
 labels_path = 'labels.json'
-
+import os
+import base64
 
 with open(labels_path) as json_data:
     idx_to_labels = json.load(json_data)
@@ -87,7 +88,51 @@ class FineTuneModel(pl.LightningModule):
    
     
     
-    
+# Get the current working directory
+current_dir = os.getcwd()
+
+# Construct the absolute path to the logo.png file
+logo_path = os.path.join(current_dir, "logo.png")
+
+with open(logo_path, "rb") as f:
+    image_data = f.read()
+    image_base64 = base64.b64encode(image_data).decode("utf-8")
+
+# Add custom CSS for the header
+header_css = """
+<style>
+.header {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    padding: 20px;
+    font-size: 18px;
+}
+
+.header img {
+    margin-right: 10px;
+    width: 80px;
+    height: 50px;
+}
+
+.header p {
+    font-size: 14px;
+}
+</style>
+"""
+
+# Render the custom CSS
+st.markdown(header_css, unsafe_allow_html=True)
+
+# Render the header
+header_html = f"""
+<div class="header">
+    <img src='data:image/jpeg;base64,{image_base64}' alt="Logo"/>
+    <p>Disclaimer: This web app is for demonstration purposes only and not intended for commercial use. Contact: [email protected] for full solution.</p>
+</div>
+"""
+
+st.markdown(header_html, unsafe_allow_html=True)
 
 st.markdown("<h1 style='text-align: center; '>Chest Xray Diagnosis</h1>",unsafe_allow_html=True)
 
@@ -127,37 +172,6 @@ if uploaded_file is not None:
                 with st.spinner('Generating Text'):
                     generator(input_text, max_length=300, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=1)
                 st.markdown(generator(input_text, max_length=300, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=1)[0]['generated_text'])
-            
-            
-            
-            
 
-    
-        
-        
-    
-    
-    
 else:
     st.success("Please upload an image file ⚕️")
-    
-    
-    
-    
-    
-
-
-
-
-
-
-
-
-
-
-
-
-
-    
-
-