itzjunayed commited on
Commit
0d1e4cd
·
verified ·
1 Parent(s): bb56ded

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -186,7 +186,7 @@ def prediction(image_id_main, local_image_path, model_path):
186
  def predict():
187
  image_id = request.form['image_id']
188
  image_file = request.files['image']
189
- model_path = request.form['model_path']
190
 
191
  local_image_path = os.path.join("input_images", image_file.filename)
192
  os.makedirs("input_images", exist_ok=True)
 
186
  def predict():
187
  image_id = request.form['image_id']
188
  image_file = request.files['image']
189
+ model_path = "model_final.pth"
190
 
191
  local_image_path = os.path.join("input_images", image_file.filename)
192
  os.makedirs("input_images", exist_ok=True)