Update app.py

app.py

@@ -20,7 +20,7 @@ def main():
                 predictor = PredictionPipeline("temp_image.png")
                 prediction = predictor.predict()
             finally:
-                os.remove(temp_file_path)
+                os.remove("temp_image.png")
             st.title(type(prediction))
             if type(prediction) == str:
                 st.json({'prediction' : prediction})