Update app.py

app.py

@@ -83,16 +83,19 @@ def main():
     # print the calories
     st.write("Selected: **{}** Max Calories.".format(calories))
     
-    uploaded_file = st.file_uploader("Upload Files",type=['png','jpeg','jpg'])
+    uploaded_file = st.file_uploader("Upload Files", type=['png','jpeg','jpg'])
 
-    if uploaded_file!=None:
-        img=Image.open(uploaded_file)
+    loading_text = st.empty()
 
+    if uploaded_file != None:
+        loading_text.markdown("Loading...")
+        img = Image.open(uploaded_file)
         extractor = AutoFeatureExtractor.from_pretrained("Kaludi/food-category-classification-v2.0")
         model = AutoModelForImageClassification.from_pretrained("Kaludi/food-category-classification-v2.0")
-
-        inputs = extractor(img,return_tensors="pt")
+        inputs = extractor(img, return_tensors="pt")
         outputs = model(**inputs)
+        # ...
+        loading_text.empty()
         label_num=outputs.logits.softmax(1).argmax(1)
         label_num=label_num.item()