Update app.py

app.py

@@ -11,16 +11,27 @@ model = None
 def load_model():
     global model
     try:
-        logger.info("Loading model with transformers...")
+        logger.info("Loading YOUR fine-tuned model...")
         from transformers import pipeline
         
-        # Use the base model
-        model = pipeline("text-generation", model="microsoft/Phi-3.5-mini-instruct", trust_remote_code=True)
-        logger.info("✅ Model loaded successfully!")
+        # Use YOUR converted model
+        model = pipeline(
+            "text-generation", 
+            model="kacperbb/phi-3.5-hf-finetuned",
+            trust_remote_code=True
+        )
+        logger.info("✅ YOUR fine-tuned model loaded successfully!")
         return True
     except Exception as e:
-        logger.error(f"❌ Error loading model: {e}")
-        return False
+        logger.error(f"❌ Error loading your model: {e}")
+        logger.info("Falling back to base model...")
+        try:
+            model = pipeline("text-generation", model="microsoft/Phi-3.5-mini-instruct")
+            logger.info("✅ Base model loaded as fallback")
+            return True
+        except Exception as e2:
+            logger.error(f"❌ Fallback failed: {e2}")
+            return False
 
 @app.route('/generate', methods=['POST'])
 def generate_text():
@@ -48,22 +59,23 @@ def generate_text():
 @app.route('/health', methods=['GET'])
 def health():
     return jsonify({
-        "status": "healthy", 
+        "status": "healthy",
         "model_loaded": model is not None
     })
 
 @app.route('/', methods=['GET'])
 def home():
     return jsonify({
-        "message": "Phi 3.5 API is running!",
+        "message": "Your Phi 3.5 Fine-tuned API is running!",
+        "model": "kacperbb/phi-3.5-hf-finetuned",
         "endpoints": {
-            "generate": "POST /generate", 
+            "generate": "POST /generate",
             "health": "GET /health"
         }
     })
 
 if __name__ == '__main__':
-    logger.info("Starting Phi 3.5 API...")
+    logger.info("Starting YOUR Phi 3.5 API...")
     load_model()
     port = int(os.environ.get('PORT', 7860))
     app.run(host='0.0.0.0', port=port, debug=False)