Spaces:
Sleeping
Sleeping
Commit
·
472e9bb
1
Parent(s):
fdf5004
temp
Browse files
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 |
-
|
| 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:
|