Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ class Prediction:
|
|
13 |
app = fastapi.FastAPI()
|
14 |
|
15 |
# Load the .bin model
|
16 |
-
model = torch.load("/main/
|
17 |
|
18 |
# Define a function to preprocess the input image
|
19 |
def preprocess_input(input: Any):
|
|
|
13 |
app = fastapi.FastAPI()
|
14 |
|
15 |
# Load the .bin model
|
16 |
+
model = torch.load("/main/model67.bin", map_location='cpu')
|
17 |
|
18 |
# Define a function to preprocess the input image
|
19 |
def preprocess_input(input: Any):
|