Update app.py

app.py

@@ -2,12 +2,16 @@ import gradio as gr
 import pandas as pd
 import torch
 from transformers import AutoTokenizer, AutoModelForMaskedLM
+import logging
 
-# Initialize tokenizer and model globally
+logging.getLogger("transformers.modeling_utils").setLevel(logging.ERROR)
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print(f"Using device: {device}")
+
+# Load the tokenizer and model
 model_name = "ChatterjeeLab/FusOn-pLM"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForMaskedLM.from_pretrained(model_name, trust_remote_code=True)
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 model.eval()