Upload app.py with huggingface_hub
Browse files
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)
|