Update app.py

app.py

@@ -5,14 +5,16 @@ from transformers import pipeline
 from PIL import Image
 import time
 
-
-'Starting a long computation...'
 pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
 st.title("Hot Dog? Or Not?")
 
 file_name = st.file_uploader("Upload a hot dog candidate image")
 
+# Add a placeholder
+latest_iteration = st.empty()
+bar = st.progress(0)
+
 if file_name is not None:
     col1, col2 = st.columns(2)
 
@@ -21,20 +23,9 @@ if file_name is not None:
     predictions = pipeline(image)
 
     col2.header("Probabilities")
-    for p in predictions:
-        col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")
-
-
-# Add a placeholder
-latest_iteration = st.empty()
-bar = st.progress(0)
-
-for i in range(100):
-  # Update the progress bar with each iteration.
-  latest_iteration.text(f'Iteration {i+1}')
-  bar.progress(i + 1)
-  time.sleep(0.1)
-
-
 
-'...and now we\'re done!'
+    i = 0
+    for p in predictions:  
+        bar.progress(i + 1)
+        time.sleep(0.1)
+        col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")