SANJAYV10 commited on
Commit
fa91d9c
1 Parent(s): 632b6a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -12,8 +12,7 @@ class Prediction:
12
 
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):
 
12
 
13
  app = fastapi.FastAPI()
14
 
15
+ model = torch.load("model67.bin", map_location='cpu')
 
16
 
17
  # Define a function to preprocess the input image
18
  def preprocess_input(input: Any):