Update app.py

app.py

@@ -132,6 +132,7 @@ async def fine_tune(params: FineTuneRequest):
     """Fine-tuning endpoint"""
     try:
         # Load datasets
+        print("Starting fine tuning..")
         alpaca = load_dataset("gbharti/finance-alpaca", split=f"train[:{params.samples_per_dataset}]")
         fiqa = load_dataset("bilalRahib/fiqa-personal-finance-dataset", "full", split=f"train[:{params.samples_per_dataset}]")
 
@@ -193,14 +194,16 @@ async def fine_tune(params: FineTuneRequest):
         )
         
         trainer.train()
+        print("Training completed, saving model...")
         model.save_pretrained("/app/finetuned")
-        
+        print("Model saved!")  # Debug log
         return {
             "status": "success",
             "trained_samples": len(dataset),
             "training_time": datetime.now().isoformat()
         }
     except Exception as e:
+        print(f"Fine-tuning failed: {e}")  # Debug log
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/health")