Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ import gradio as gr
|
|
20 |
|
21 |
|
22 |
|
23 |
-
model = torch.jit.load('
|
24 |
|
25 |
model.eval()
|
26 |
|
|
|
20 |
|
21 |
|
22 |
|
23 |
+
model = torch.jit.load('Net2_Flip_jit.pt', map_location = torch.device('cpu'))
|
24 |
|
25 |
model.eval()
|
26 |
|