Update app.py

app.py

@@ -7,6 +7,7 @@ import gradio as gr
 import cv2
 import numpy as np
 import scipy
+import base64
 
 model_path = "pretrained_models/model_qnrf.pth"
 url = "https://drive.google.com/uc?id=1nnIHPaV9RGqK8JHL645zmRvkNrahD9ru"
@@ -20,6 +21,10 @@ model.load_state_dict(torch.load(model_path, device))
 model.eval()
 
 
+def image_to_base64(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode('utf-8')
+
 def predict(inp):
     inp = Image.fromarray(inp.astype('uint8'), 'RGB')
     inp = transforms.ToTensor()(inp).unsqueeze(0)
@@ -42,13 +47,41 @@ outputs = [
     gr.Label(label="Predicted Count")
 ]
 
-# Assuming `title`, `desc`, and `examples` variables are defined elsewhere in your code.
-title = "Your App Title"
-desc = "Your App Description"
+# Convert your image to Base64
+logo_base64 = image_to_base64("corporate.png")
+logo_src = f"data:image/png;base64,{logo_base64}"
+
+
+desc = f"""
+<style>
+/* Add padding at the bottom of the interface to prevent overlap with the absolutely positioned image */
+body, html, .interface {{
+    margin: 0;
+    padding-bottom: 120px; /* Adjust this value based on the height of your image */
+}}
+/* Position your image at the bottom left of the interface */
+.logo-img {{
+    width: 150px; /* Set width to auto to keep the original image size */
+    bottom: 10px;
+    left: 10px;
+}}
+/* Style for your text to make sure it does not overlap the logo */
+.with-margin {{
+    padding-left: 120px; /* This padding should be more than the width of your logo to prevent text overlap */
+}}
+</style>
+<div class="description">
+    <img src="{logo_src}" alt='Logo' class="logo-img" width="300px"/>
+    <h4 class="with-margin">AI-Powered Audience Insights</h4>
+    <p class="with-margin">Seamlessly count and analyze your conference attendees with cutting-edge neural language models</p>
+</div>
+"""
+
 
 gr.Interface(fn=predict, 
              inputs=inputs, 
-             outputs=outputs, 
-             title=title, 
+             outputs=outputs,
+             title=" ", 
              description=desc, 
-             allow_flagging="never").launch(share=True)
+             allow_flagging="never",
+             css="footer{display:none !important}").launch()