Spaces:
Runtime error
Runtime error
model67 replaced with best_model.pth
Browse files
app.py
CHANGED
|
@@ -11,7 +11,8 @@ class Prediction:
|
|
| 11 |
|
| 12 |
app = fastapi.FastAPI(docs_url="/")
|
| 13 |
|
| 14 |
-
model = torch.load("model67.bin", map_location='cpu')
|
|
|
|
| 15 |
print(model)
|
| 16 |
# Define a function to preprocess the input image
|
| 17 |
def preprocess_input(input: fastapi.UploadFile):
|
|
|
|
| 11 |
|
| 12 |
app = fastapi.FastAPI(docs_url="/")
|
| 13 |
|
| 14 |
+
# model = torch.load("model67.bin", map_location='cpu')
|
| 15 |
+
model = torch.load("best_model.pth", map_location='cpu')
|
| 16 |
print(model)
|
| 17 |
# Define a function to preprocess the input image
|
| 18 |
def preprocess_input(input: fastapi.UploadFile):
|