liuganghuggingface commited on
Commit
7173a2e
·
verified ·
1 Parent(s): 80330c5

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -54,13 +54,14 @@ def random_properties():
54
 
55
  # def load_model(model_choice):
56
  # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
57
 
58
  model_all = load_graph_decoder(path='model_all')
59
  model_labeled = load_graph_decoder(path='model_labeled')
60
- # return (model, device)
61
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
62
  print('device')
63
 
 
64
  # Create a flagged folder if it doesn't exist
65
  flagged_folder = "flagged"
66
  os.makedirs(flagged_folder, exist_ok=True)
 
54
 
55
  # def load_model(model_choice):
56
  # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
57
+ # return (model, device)
58
 
59
  model_all = load_graph_decoder(path='model_all')
60
  model_labeled = load_graph_decoder(path='model_labeled')
 
61
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
62
  print('device')
63
 
64
+
65
  # Create a flagged folder if it doesn't exist
66
  flagged_folder = "flagged"
67
  os.makedirs(flagged_folder, exist_ok=True)