Update api.py

api.py

@@ -1,8 +1,17 @@
+import time
 from flask import Flask, request, jsonify
 from app import generate_text
 
 app = Flask(__name__)
 
+# Warm up the model on startup
+print("Warming up model...")
+try:
+    generate_text("Warm up", max_length=10)
+    print("Model ready!")
+except Exception as e:
+    print(f"Warmup failed: {str(e)}")
+
 @app.route("/generate", methods=["POST"])
 def generate():
     data = request.json