add logo

app.py

@@ -2,6 +2,9 @@ import streamlit as st
 import torch
 import torch.nn as nn
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import requests
+from PIL import Image
+from io import BytesIO
 
 # Define the template
 TEMPLATE = """[INST] Evaluate if the given text is harmful i.e. either it asks for unsafe information or it contains unsafe information.
@@ -21,6 +24,13 @@ def load_model():
 
 tokenizer, model = load_model()
 
+# Function to load image from URL
+@st.cache_data
+def load_image_from_url(url):
+    response = requests.get(url)
+    img = Image.open(BytesIO(response.content))
+    return img
+
 # Streamlit app
 st.title("Text Safety Evaluator")
 
@@ -57,4 +67,6 @@ st.write("Connect with us at admin@walled.ai for more information.")
 # Add logo at the bottom center
 col1, col2, col3 = st.columns([1,2,1])
 with col2:
-    st.image("logo.png", use_column_width=True)
+    logo_url = "https://github.com/walledai/walledeval/assets/32847115/d8b1d14f-7071-448b-8997-2eeba4c2c8f6"
+    logo = load_image_from_url(logo_url)
+    st.image(logo, use_column_width=True)