Update app.py

app.py

@@ -13,7 +13,7 @@ computer_images = {
     "scissors": "computer_scissors.png"
 }
 
-# טעינת מודל
+# טעינת המודל
 checkpoint = "facebook/deit-tiny-patch16-224"
 processor = AutoImageProcessor.from_pretrained(checkpoint)
 model = AutoModelForImageClassification.from_pretrained(
@@ -24,7 +24,7 @@ model = AutoModelForImageClassification.from_pretrained(
     ignore_mismatched_sizes=True
 )
 
-# חוקים
+# חוקי המשחק
 def game_logic(user_move, computer_move):
     if user_move == computer_move:
         return "It's a tie!"
@@ -35,8 +35,14 @@ def game_logic(user_move, computer_move):
     else:
         return "You lose!"
 
-# פונקציית משחק מלאה
-def play(live_image):
+# ספירה לאחור + משחק
+def full_play(live_image):
+    # ספירה לאחור בשרת (יופיע בלוגים)
+    for i in ["3...", "2...", "1...", "GO!"]:
+        print(i)
+        time.sleep(1)
+
+    # ניתוח תמונה
     prediction = processor(images=live_image, return_tensors="pt")
     outputs = model(**prediction)
     logits = outputs.logits
@@ -50,19 +56,11 @@ def play(live_image):
 
     return live_image, computer_img, f"You chose {user_move}, computer chose {computer_move}. {result}"
 
-# ספירה לאחור + משחק
-def full_play(live_image):
-    # ספירה לאחור
-    for i in ["3...", "2...", "1...", "GO!"]:
-        print(i)
-        time.sleep(1)
-    return play(live_image)
-
-# בניית אפליקציה
+# בניית האפליקציה
 with gr.Blocks() as demo:
     gr.Markdown("# ✂️ 🪨 📄 Rock Paper Scissors - LIVE Game!")
 
-    webcam = gr.Camera(label="Show your move after countdown!")
+    webcam_input = gr.Image(source="webcam", tool=None, label="Show your move!")
     play_button = gr.Button("Start Countdown and Play")
 
     user_output = gr.Image(label="Your Move")
@@ -71,7 +69,7 @@ with gr.Blocks() as demo:
 
     play_button.click(
         fn=full_play,
-        inputs=[webcam],
+        inputs=[webcam_input],
         outputs=[user_output, computer_output, result_text]
     )