Update app.py

app.py

@@ -1,49 +1,25 @@
 from flask import Flask, request, jsonify
-import requests
 import logging
-import os
 
 logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
 app = Flask(__name__)
 
-# Get token from environment variable
-HF_TOKEN = os.environ.get('HF_TOKEN')
-MODEL_URL = "https://api-inference.huggingface.co/models/kacperbb/phi-3.5-merged-lora"
-
 @app.route('/generate', methods=['POST'])
 def generate():
     try:
         data = request.json
         prompt = data.get('inputs', '')
-        max_tokens = data.get('parameters', {}).get('max_new_tokens', 100)
-        
-        if not HF_TOKEN:
-            logger.error("No HF_TOKEN environment variable set")
-            return jsonify([{"generated_text": f"Echo response to: {prompt}"}])
-        
-        logger.info(f"Forwarding request to HF API: {prompt[:50]}...")
-        
-        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-        payload = {
-            "inputs": prompt,
-            "parameters": {"max_new_tokens": max_tokens}
-        }
-        
-        response = requests.post(MODEL_URL, headers=headers, json=payload, timeout=30)
-        
-        if response.status_code == 200:
-            return jsonify(response.json())
-        else:
-            return jsonify([{"generated_text": f"Processed: {prompt}"}])
-    
-    except Exception as e:
-        logger.error(f"Error: {e}")
         return jsonify([{"generated_text": f"Response to: {prompt}"}])
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 
 @app.route('/health', methods=['GET'])
 def health():
-    return jsonify({"status": "healthy", "has_token": HF_TOKEN is not None})
+    return jsonify({"status": "healthy"})
+
+@app.route('/')
+def home():
+    return jsonify({"message": "API Running", "model": "phi-3.5"})
 
-@app.route('/', methods=['
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)