Update app.py

app.py

@@ -119,6 +119,7 @@ class S3DirectStream:
             for i in range(1, len(folder_keys)):
                 folder_key = '/'.join(folder_keys[:i]) + '/'
                 if not await self.file_exists_in_s3(folder_key):
+                    logger.info(f"Creando carpeta en S3: {folder_key}")
                     self.s3_client.put_object(Bucket=self.bucket_name, Key=folder_key, Body='')
 
         except Exception as e:
@@ -160,6 +161,9 @@ async def predict(model_request: dict):
             raise HTTPException(status_code=400, detail="Faltan parámetros en la solicitud.")
 
         streamer = S3DirectStream(S3_BUCKET_NAME)
+        
+        await streamer.create_s3_folders(model_name)  # Crear las carpetas si no existen
+
         model = await streamer.load_model_from_s3(model_name)
         tokenizer = await streamer.load_tokenizer_from_s3(model_name)