update error printing

Dockerfile

@@ -24,4 +24,4 @@ ENV HOME=/home/user \
 EXPOSE 7860 
 
 # Command to run the application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--interface", "wsgi"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--interface", "wsgi", "--log-level", "debug"]

app.py

@@ -1,5 +1,6 @@
 from flask import Flask, render_template, request, jsonify
 import numpy as np
+import traceback  # <--- Add this import
 import torch
 
 import sys
@@ -68,14 +69,19 @@ def index():
 
 @app.route('/process_sentence', methods=['POST'])
 def process_sentence():
-    req_data = request.json
-    sentence = req_data.get('sentence', '')
-    layer = int(req_data.get('layer', 5))
-    head = int(req_data.get('head', 0))
-
-    # Calculate data
-    results = model.forward(sentence, layer, head)
-    return jsonify({"status": "success", "data": results})
+    try:
+        req_data = request.json
+        sentence = req_data.get('sentence', '')
+        layer = int(req_data.get('layer', 5))
+        head = int(req_data.get('head', 0))
+        # Calculate data
+        results = model.forward(sentence, layer, head)
+        return jsonify({"status": "success", "data": results})
+    except Exception as e:
+        # 1. Print full error to the Server Logs (so you can see it in HF console)
+        print("Error processing request:", flush=True)
+        print(traceback.format_exc(), flush=True)
+        
 
 # if __name__ == '__main__':
 #     _test_repo()