chore: cleanup

pull/160/head
Zack 1 year ago
parent 4e63bad703
commit b5dbc01527

@ -1,6 +1,4 @@
import json
import os import os
import sys
import time import time
from functools import partial from functools import partial
from pathlib import Path from pathlib import Path
@ -10,6 +8,8 @@ import warnings
from swarms.modelui.modules.block_requests import OpenMonkeyPatch, RequestBlocker from swarms.modelui.modules.block_requests import OpenMonkeyPatch, RequestBlocker
from swarms.modelui.modules.logging_colors import logger from swarms.modelui.modules.logging_colors import logger
from vllm import LLM, SamplingParams
os.environ['GRADIO_ANALYTICS_ENABLED'] = 'False' os.environ['GRADIO_ANALYTICS_ENABLED'] = 'False'
os.environ['BITSANDBYTES_NOWELCOME'] = '1' os.environ['BITSANDBYTES_NOWELCOME'] = '1'
warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated') warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated')
@ -48,12 +48,10 @@ from swarms.modelui.modules.models_settings import (
) )
from swarms.modelui.modules.utils import gradio from swarms.modelui.modules.utils import gradio
import yaml
import gradio as gr import gradio as gr
from swarms.tools.tools_controller import MTQuestionAnswerer, load_valid_tools from swarms.tools.tools_controller import MTQuestionAnswerer, load_valid_tools
from swarms.tools.singletool import STQuestionAnswerer from swarms.tools.singletool import STQuestionAnswerer
from langchain.schema import AgentFinish from langchain.schema import AgentFinish
import os
import requests import requests
from swarms.modelui.server import create_interface from swarms.modelui.server import create_interface
@ -274,9 +272,6 @@ with gr.Blocks() as demo:
with gr.Column(scale=1): with gr.Column(scale=1):
gr.Image(show_label=False, show_download_button=False, value="images/swarmslogobanner.png") gr.Image(show_label=False, show_download_button=False, value="images/swarmslogobanner.png")
with gr.Tab("Models"):
create_interface()
with gr.Tab("Key setting"): with gr.Tab("Key setting"):
OPENAI_API_KEY = gr.Textbox(label="OpenAI API KEY:", placeholder="sk-...", type="text") OPENAI_API_KEY = gr.Textbox(label="OpenAI API KEY:", placeholder="sk-...", type="text")
WOLFRAMALPH_APP_ID = gr.Textbox(label="Wolframalpha app id:", placeholder="Key to use wlframalpha", type="text") WOLFRAMALPH_APP_ID = gr.Textbox(label="Wolframalpha app id:", placeholder="Key to use wlframalpha", type="text")

Loading…
Cancel
Save