Update app.py

app.py

@@ -61,7 +61,7 @@ import gc
 import os
 
 config = {
-    'model': 'kaitehtzeng/primary_app/microsoft/deberta-v3-base',
+    'model': 'microsoft/deberta-v3-base',
     'dropout': 0.2,
     'max_length': 512,
     'batch_size':3,
@@ -76,7 +76,7 @@ Comparing two essays. <br>
 One predicted written by students, one predicted written by LLM
 """
 
-train_essays = pd.read_csv("kaitehtzeng/primary_app/train_essays.csv")
+train_essays = pd.read_csv("train_essays.csv")
 
 
 import transformers
@@ -133,7 +133,7 @@ class MeanPooling(nn.Module):
         return mean_embedding
 
 model = mymodel(config).to(device=config['device'])
-model.load_state_dict(torch.load('kaitehtzeng/primary_app/my_model.pth'))
+model.load_state_dict(torch.load('my_model.pth'))
 model.eval()
 
 #preds = []