Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ csv.field_size_limit(sys.maxsize)
19
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
20
 
21
  # Load models and tokenizer
22
- tokenizer = AutoTokenizer.from_pretrained("pxf4fp/AwareRx")
23
- model = AutoModelForSequenceClassification.from_pretrained("pxf4fp/AwareRx").to(device)
24
 
25
  # Prediction pipeline
26
  pred = transformers.pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=None, device=device)
 
19
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
20
 
21
  # Load models and tokenizer
22
+ tokenizer = AutoTokenizer.from_pretrained("TyHamil/ADRv2025")
23
+ model = AutoModelForSequenceClassification.from_pretrained("TyHamil/ADRv2025").to(device)
24
 
25
  # Prediction pipeline
26
  pred = transformers.pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=None, device=device)