httpsAkayush commited on
Commit
fdf5004
·
1 Parent(s): c471abc
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -86,7 +86,7 @@ async def predict_disease(file: UploadFile = File(...)):
86
  # Validate file type
87
 
88
  # Save uploaded file temporarily
89
- with tempfile.NamedTemporaryFile(suffix=".jpg", delete=True) as tmp:
90
  temp_path = tmp.name
91
  tmp.write(await file.read())
92
  tmp.flush() # Ensure data is written
@@ -108,6 +108,8 @@ async def predict_disease(file: UploadFile = File(...)):
108
  confidence = prediction[0][result_index]
109
  disease_name = class_name[result_index]
110
 
 
 
111
  return {
112
  "success": True,
113
  "disease": disease_name,
 
86
  # Validate file type
87
 
88
  # Save uploaded file temporarily
89
+ with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
90
  temp_path = tmp.name
91
  tmp.write(await file.read())
92
  tmp.flush() # Ensure data is written
 
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,