Update app.py

app.py

@@ -19,8 +19,8 @@ csv.field_size_limit(sys.maxsize)
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
-tokenizer = AutoTokenizer.from_pretrained("MarkAdamsMSBA24/ADRv2024")  
-model = AutoModelForSequenceClassification.from_pretrained("MarkAdamsMSBA24/ADRv2024").to(device)
+tokenizer = AutoTokenizer.from_pretrained("vikvenk/ADR_Detection")  
+model = AutoModelForSequenceClassification.from_pretrained("vikvenk/ADR_Detection").to(device)
 
 # build a pipeline object to do predictions
 pred = transformers.pipeline("text-classification", model=model,