httpsAkayush commited on
Commit
472e9bb
·
1 Parent(s): fdf5004
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -108,13 +108,18 @@ async def predict_disease(file: UploadFile = File(...)):
108
  confidence = prediction[0][result_index]
109
  disease_name = class_name[result_index]
110
 
111
- os.remove(temp_path)
112
 
113
  return {
114
  "success": True,
115
  "disease": disease_name,
116
  "confidence": confidence
117
  }
 
 
 
 
 
118
 
119
 
120
  except Exception as e:
 
108
  confidence = prediction[0][result_index]
109
  disease_name = class_name[result_index]
110
 
111
+
112
 
113
  return {
114
  "success": True,
115
  "disease": disease_name,
116
  "confidence": confidence
117
  }
118
+
119
+ finally:
120
+ # Delete the temporary file manually
121
+ if os.path.exists(temp_path):
122
+ os.remove(temp_path)
123
 
124
 
125
  except Exception as e: