initial commit

app.py

@@ -11,7 +11,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model, tokenizer = None, None
 
 
-@spaces.GPU
 def load_model_tokenizer(checkpoint):
     global model, tokenizer
     model = AutoModelForSequenceClassification.from_pretrained(checkpoint)
@@ -23,6 +22,7 @@ def load_model_tokenizer(checkpoint):
 load_model_tokenizer(model_checkpoint)
 
 
+@spaces.GPU
 def prediction(text):
     encoded_text = tokenizer(text, return_tensors="pt").to(device)
     with torch.inference_mode():