Spaces:

bvd757
/

Papers_classification

Running

bvd757 commited on Apr 8

Commit

1eeb83f

1 Parent(s): 283dd4a

app fixes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ def forward(self,
 def load_model(test=False):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     try:
-        path = 'Homework4'
         with open(f'{path}/label_to_theme.json', 'r') as f:
             label_to_theme = json.load(f)
     except:

 def load_model(test=False):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     try:
+        path = 'Papers_classification'
         with open(f'{path}/label_to_theme.json', 'r') as f:
             label_to_theme = json.load(f)
     except: