Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import re
 import json
 import requests
-import torch
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from google.cloud import storage
@@ -26,6 +25,12 @@ def validate_bucket_name(bucket_name):
         raise ValueError(f"El nombre del bucket '{bucket_name}' no es válido. Debe comenzar y terminar con una letra o número.")
     return bucket_name
 
+# Validar nombre del repositorio en Hugging Face
+def validate_huggingface_repo_name(repo_name):
+    if not re.match(r"^[a-zA-Z0-9_.-]+$", repo_name):
+        raise ValueError(f"El nombre del repositorio '{repo_name}' no es válido. Debe contener solo letras, números, '-', '_', y '.'")
+    return repo_name
+
 # Inicialización del cliente de GCS
 try:
     GCS_BUCKET_NAME = validate_bucket_name(GCS_BUCKET_NAME)  # Validar el nombre del bucket
@@ -88,6 +93,8 @@ def download_model_from_huggingface(model_name):
     """
     Descarga un modelo desde Hugging Face y lo sube a GCS en streaming.
     """
+    model_name = validate_huggingface_repo_name(model_name)  # Validar nombre del repositorio
+
     file_patterns = [
         "pytorch_model.bin",
         "model.safetensors",