Update app.py

app.py

@@ -2,12 +2,15 @@ import gradio as gr
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
+# Select the appropriate device
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 # Load the model and tokenizer
-model = AutoModelForSequenceClassification.from_pretrained("./data", local_files_only=True)
+model = AutoModelForSequenceClassification.from_pretrained("./", local_files_only=True).to(device)
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
 
 def classify_text(text):
-    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
+    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512).to(device)
     with torch.no_grad():
         outputs = model(**inputs)
     logits = outputs.logits