Update app.py

app.py

@@ -139,6 +139,13 @@ class S3DirectStream:
             logger.error(f"Error al subir archivo a S3: {e}")
             raise HTTPException(status_code=500, detail=f"Error al subir archivo a S3: {e}")
 
+    def file_exists_in_s3(self, s3_key):
+        try:
+            self.s3_client.head_object(Bucket=self.bucket_name, Key=s3_key)
+            return True
+        except self.s3_client.exceptions.ClientError:
+            return False
+
 @app.post("/predict/")
 async def predict(model_request: dict):
     try:
@@ -146,6 +153,9 @@ async def predict(model_request: dict):
         task = model_request.get("pipeline_task")
         input_text = model_request.get("input_text")
 
+        if not model_name or not task or not input_text:
+            raise HTTPException(status_code=400, detail="Faltan parámetros en la solicitud.")
+
         streamer = S3DirectStream(S3_BUCKET_NAME)
         model = streamer.load_model_from_s3(model_name)
         tokenizer = streamer.load_tokenizer_from_s3(model_name)