Update app.py

app.py

@@ -342,6 +342,9 @@ def run_tuning_task():
         liked_data = feedback_data.filter(lambda x: x['feedback'] == 1)
         print(f"Found {len(liked_data)} total liked responses for training.")
 
+        # Add shuffle and sample 10,000 random examples
+        liked_data = liked_data.shuffle(seed=42).select(range(10000))  # Use first 10,000 samples
+        
         if len(liked_data) < MIN_LIKES_FOR_TRAINING:
             TRAINING_STATUS = f"✅ Improvement complete! (Not enough new data to train, will try again later)."
             time.sleep(5)
@@ -452,6 +455,7 @@ def run_tuning_task():
         training_lock.release()
         print("--- PyTorch Fine-Tuning Task Finished ---")
 
+
 # -------------------------------
 # 7) UI Functions & Gradio Interface
 # -------------------------------
@@ -482,6 +486,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="blue")) a
     chatbot.like(log_feedback, inputs=[chatbot], outputs=None)
     
     demo.load(poll_status_updater, None, training_status_md)
+
+    
 if __name__ == "__main__":
     print("Starting Gradio app. Tuning will be triggered by user feedback.")
     demo.launch(show_api=True)