Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 import numpy as np
 import io
 import time
+import math
 
 # Hugging Face Transformers specific imports
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
@@ -33,15 +34,25 @@ def main():
         image = Image.open(uploaded_file)
         st.image(image, caption="Uploaded Image", use_column_width=True)
 
-        # Rotate the image at a constant speed
-        st.write("Rotating image...")
+        # Rotation angle initialization
         rotation_angle = 0
+
+        # Display an empty placeholder
+        rotating_image_placeholder = st.empty()
+
         while True:
-            rotated_image = rotate_image(image, rotation_angle)
-            st.image(rotated_image, caption="Rotated Image", use_column_width=True)
-            time.sleep(0.1)  # Adjust rotation speed
+            # Calculate the new rotation angle
             rotation_angle += rotation_speed
             rotation_angle %= 360  # Keep angle within 0-359 range
 
+            # Rotate the image
+            rotated_image = rotate_image(image, rotation_angle)
+
+            # Display the rotated image
+            rotating_image_placeholder.image(rotated_image, caption="Rotated Image", use_column_width=True)
+
+            # Pause briefly to control the rotation speed
+            time.sleep(0.1)
+
 if __name__ == "__main__":
     main()