Upload folder using huggingface_hub

main.py

@@ -13,9 +13,17 @@ app = FastAPI(title="BERT Attention Visualizer Backend")
 # Ensure the server correctly identifies forwarded HTTPS requests
 @app.middleware("http")
 async def force_https(request: Request, call_next):
-    if request.headers.get("x-forwarded-proto") == "http":
+    # Check if request is HTTP and needs to be upgraded to HTTPS
+    forwarded_proto = request.headers.get("x-forwarded-proto")
+    
+    if forwarded_proto == "http":
         # Redirect to HTTPS if accessed via HTTP
-        return RedirectResponse(url=f"https://{request.headers['host']}{request.url.path}")
+        https_url = f"https://{request.headers['host']}{request.url.path}"
+        if request.query_params:
+            https_url += f"?{request.query_params}"
+        return RedirectResponse(url=https_url, status_code=301)  # Permanent redirect
+        
+    # Continue with the request if already HTTPS or no proto header
     return await call_next(request)
 
 # Restrict CORS to your frontend

routes/attention.py

@@ -4,7 +4,7 @@ from helpers import *
 from routes.tokenize import tokenize_text
 router = APIRouter()
 
-@router.post("/", response_model=AttentionResponse)
+@router.post("", response_model=AttentionResponse)
 async def get_attention_matrices(request: AttentionRequest):
     """Get attention matrices for the input text using the specified model"""
     try:

routes/attention_comparison.py

@@ -5,7 +5,7 @@ from attention_comparison_helpers import *
 router = APIRouter()
 
 
-@router.post("/", response_model=AttentionComparisonResponse)
+@router.post("", response_model=AttentionComparisonResponse)
 async def get_attention_comparison(request: ComparisonRequest):
     """
     Dispatcher for attention comparison - routes to the appropriate model-specific implementation

routes/mask_prediction.py

@@ -7,7 +7,7 @@ router = APIRouter()
 
 
 
-@router.post("/", response_model=MaskPredictionResponse)
+@router.post("", response_model=MaskPredictionResponse)
 async def predict_masked_token(request: MaskPredictionRequest, x_token_to_mask: str = Header(None), x_explicit_masked_text: str = Header(None)):
     """Predict masked token using the specified model"""
     try:

routes/models.py

@@ -3,7 +3,7 @@ from models import MODEL_CONFIGS
 router = APIRouter()
 
 
-@router.get("/")
+@router.get("")
 async def get_available_models():
     """Get list of available models"""
     return {

routes/tokenize.py

@@ -4,7 +4,7 @@ from helpers import *
 
 router = APIRouter()
 
-@router.post("/", response_model=TokenizeResponse)
+@router.post("", response_model=TokenizeResponse)
 async def tokenize_text(request: TokenizeRequest):
     """Tokenize input text using the specified model's tokenizer"""
     try: