Spaces:

pr4nav101
/

Traffic_Optimization

Paused

pr4nav101 commited on May 21, 2024

Commit

7700efe

verified ·

1 Parent(s): 570983a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -169,7 +169,7 @@ elif page == "PS3":
     model2 = YOLO("best.pt")
     if uploaded_file1 is not None:
         with tempfile.NamedTemporaryFile(delete=False) as temp:
-            temp.write(uploaded_file.read())
             if 'roi_list2' not in st.session_state:
                 st.session_state['roi_list2'] = []
             if "all_rois2" not in st.session_state:

     model2 = YOLO("best.pt")
     if uploaded_file1 is not None:
         with tempfile.NamedTemporaryFile(delete=False) as temp:
+            temp.write(uploaded_file1.read())
             if 'roi_list2' not in st.session_state:
                 st.session_state['roi_list2'] = []
             if "all_rois2" not in st.session_state: