Scezui commited on
Commit
6ffd275
·
1 Parent(s): 18a4e9d
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -24,6 +24,7 @@ import signal
24
  import shutil
25
  from datetime import datetime
26
  import zipfile
 
27
 
28
  # LLM
29
  import argparse
@@ -101,19 +102,21 @@ def upload_files():
101
  return render_template('index.html')
102
 
103
 
104
- from pathlib import Path
105
- from fastai.vision import load_learner, open_image
106
-
107
  def make_predictions(image_paths):
 
108
  try:
109
- model_path = Path('model/export')
 
 
 
 
110
  learner = load_learner(model_path)
111
 
112
  predictions = []
113
 
114
  for image_path in image_paths:
115
  # Open the image using fastai's open_image function
116
- image = open_image(Path(image_path))
117
 
118
  # Make a prediction
119
  prediction_class, prediction_idx, probabilities = learner.predict(image)
@@ -127,6 +130,9 @@ def make_predictions(image_paths):
127
 
128
  except Exception as e:
129
  return {"error in make_predictions": str(e)}
 
 
 
130
 
131
 
132
  @app.route('/predict/<filenames>', methods=['GET', 'POST'])
 
24
  import shutil
25
  from datetime import datetime
26
  import zipfile
27
+ from pathlib import Path
28
 
29
  # LLM
30
  import argparse
 
102
  return render_template('index.html')
103
 
104
 
 
 
 
105
  def make_predictions(image_paths):
106
+ temp = None
107
  try:
108
+ # For Windows OS
109
+ temp = pathlib.PosixPath # Save the original state
110
+ pathlib.PosixPath = pathlib.WindowsPath # Change to WindowsPath temporarily
111
+
112
+ model_path = Path(r'model/export')
113
  learner = load_learner(model_path)
114
 
115
  predictions = []
116
 
117
  for image_path in image_paths:
118
  # Open the image using fastai's open_image function
119
+ image = open_image(image_path)
120
 
121
  # Make a prediction
122
  prediction_class, prediction_idx, probabilities = learner.predict(image)
 
130
 
131
  except Exception as e:
132
  return {"error in make_predictions": str(e)}
133
+
134
+ finally:
135
+ pathlib.PosixPath = temp
136
 
137
 
138
  @app.route('/predict/<filenames>', methods=['GET', 'POST'])