Update app.py

app.py

@@ -63,7 +63,7 @@ uploaded_file = st.file_uploader("", type=['jpg', 'png', 'jpeg', 'mp4'])
 if uploaded_file is not None:
     file_extension = os.path.splitext(uploaded_file.name)[1].lower()
 
-    if file_extension in ['jpg', 'png', 'jpeg']:
+    if file_extension in ['.jpg', '.png', '.jpeg']:
         image = Image.open(uploaded_file)
 
         col1, col2 = st.columns([0.5, 0.5])
@@ -90,7 +90,8 @@ if uploaded_file is not None:
                 st.image(out_img_siggraph17, width=300)
             else:
                 st.image(image, width=300)
-    elif file_extension == 'mp4':  # If uploaded file is a video
+                
+    elif file_extension == '.mp4':  # If uploaded file is a video
         # Save the video file to a temporary location
         temp_file = tempfile.NamedTemporaryFile(delete=False)
         temp_file.write(uploaded_file.read())