Fix data saving

backend/server.py

@@ -418,7 +418,7 @@ def run_pipeline():
         _GENERATION_RESULTS = evaluated_results
 
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        os.makedirs("data", exist_ok=True)
+        os.makedirs("tmp", exist_ok=True)
         output_file = f"/tmp/pipeline_generation_{timestamp}.csv"
         evaluated_results.to_csv(output_file, index=False)
         results['generation_file'] = output_file
@@ -589,7 +589,7 @@ def run_pipeline():
 
 
 if __name__ == '__main__':
-    os.makedirs("data", exist_ok=True)
+    os.makedirs("tmp", exist_ok=True)
     print("Starting minimal Flask server...")
     print("Available endpoints:")
     print("  GET  /health - Health check")

backend/utils/utils.py

@@ -102,7 +102,7 @@ def load_model_and_tokenizer(model_name: str):
         print(f"Traceback: {traceback.format_exc()}")
         raise RuntimeError(f"Failed to load model '{model_name}': {e}")
     
-def finetune(train_texts, tokenizer, model, num_epochs=20, output_dir='./data'):
+def finetune(train_texts, tokenizer, model, num_epochs=20, output_dir='/temp/'):
     train_path = f"/tmp/train.txt"
 
     with open(train_path, "w", encoding="utf-8") as f:
@@ -575,7 +575,7 @@ def _ensure_plot_saved(
     target: float = None,
     bins: int = 30,
 ) -> str:
-    os.makedirs("data", exist_ok=True)
+    os.makedirs("tmp", exist_ok=True)
     path = os.path.join("data", f"{basename}.png")
 
     plt.figure(figsize=(8, 5))