Spaces:
Runtime error
Runtime error
| import os | |
| from pathlib import Path | |
| from typing import Dict, List | |
| from fastapi.templating import Jinja2Templates | |
| from swarms.agents.utils.agent_creator import AgentManager | |
| from swarms.utils.main import BaseHandler, FileHandler, FileType | |
| from swarms.tools.main import ExitConversation, RequestsGet, CodeEditor, Terminal | |
| from swarms.utils.main import CsvToDataframe | |
| from swarms.tools.main import BaseToolSet | |
| from swarms.utils.main import StaticUploader | |
| BASE_DIR = Path(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | |
| os.chdir(BASE_DIR / os.environ["PLAYGROUND_DIR"]) | |
| # | |
| toolsets: List[BaseToolSet] = [ | |
| Terminal(), | |
| CodeEditor(), | |
| RequestsGet(), | |
| ExitConversation(), | |
| ] | |
| handlers: Dict[FileType, BaseHandler] = {FileType.DATAFRAME: CsvToDataframe()} | |
| if os.environ["USE_GPU"]: | |
| import torch | |
| # from core.handlers.image import ImageCaptioning | |
| from swarms.tools.main import ImageCaptioning | |
| from swarms.tools.main import ( | |
| ImageEditing, | |
| InstructPix2Pix, | |
| Text2Image, | |
| VisualQuestionAnswering, | |
| ) | |
| if torch.cuda.is_available(): | |
| toolsets.extend( | |
| [ | |
| Text2Image("cuda"), | |
| ImageEditing("cuda"), | |
| InstructPix2Pix("cuda"), | |
| VisualQuestionAnswering("cuda"), | |
| ] | |
| ) | |
| handlers[FileType.IMAGE] = ImageCaptioning("cuda") | |
| agent_manager = AgentManager.create(toolsets=toolsets) | |
| file_handler = FileHandler(handlers=handlers, path=BASE_DIR) | |
| templates = Jinja2Templates(directory=BASE_DIR / "api" / "templates") | |
| uploader = StaticUploader.from_settings( | |
| path=BASE_DIR / "static", endpoint="static" | |
| ) | |
| reload_dirs = [BASE_DIR / "core", BASE_DIR / "api"] |