Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import time
 from bs4 import BeautifulSoup
 from PIL import Image, ImageOps  
 import google.generativeai as genai
@@ -49,18 +48,17 @@ if uploaded_file is not None or photo is not None:
         img = Image.open(uploaded_file).convert("RGB")
         st.image(img, caption='Hình ảnh đã tải lên.', use_column_width=True)
 
-    data = np.ndarray(shape=(1, 224, 224, 3), dtype=np.float32)
+    with st.spinner('Đang xử lý...'):
+        data = np.ndarray(shape=(1, 224, 224, 3), dtype=np.float32)
 
-    size = (224, 224)
-    img = ImageOps.fit(img, size, Image.Resampling.LANCZOS)
+        size = (224, 224)
+        img = ImageOps.fit(img, size, Image.Resampling.LANCZOS)
 
-    image_array = np.asarray(img)
-    normalized_image_array = (image_array.astype(np.float32) / 127.5) - 1
-    data[0] = normalized_image_array
+        image_array = np.asarray(img)
+        normalized_image_array = (image_array.astype(np.float32) / 127.5) - 1
+        data[0] = normalized_image_array
 
-    with st.spinner('Đang xử lý...'):
         prediction = model.predict(data)
-        time.sleep(1)
 
     index = np.argmax(prediction)
     class_name = class_names[index]