Update app.py

app.py

@@ -1,17 +1,6 @@
 import streamlit as st
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-from PIL import Image
-import requests
-from io import BytesIO
-
-
-@st.cache_data
-def load_header_image():
-    response = requests.get(
-        "https://upload.wikimedia.org/wikipedia/commons/thumb/b/bc/ArXiv_logo_2022.svg/512px-ArXiv_logo_2022.svg.png"
-    )
-    return Image.open(BytesIO(response.content))
 
 
 @st.cache_resource
@@ -57,8 +46,6 @@ def predict(title, abstract, model, tokenizer, idx_to_category, threshold=0.95):
 
 def main():
     model, tokenizer, idx_to_category = load_model()
-    header_img = load_header_image()
-    
     st.set_page_config(page_title="arXiv Classifier", layout="wide")
     
     col1, col2 = st.columns([1, 4])