Spaces:
Running
Running
Migrate to new backend URL format
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from functools import partial
|
|
| 6 |
|
| 7 |
import gradio as gr
|
| 8 |
import httpx
|
| 9 |
-
from const import
|
| 10 |
from openai import OpenAI
|
| 11 |
from PIL import Image
|
| 12 |
|
|
@@ -90,7 +90,7 @@ def call_chat_api(message, history, model_name):
|
|
| 90 |
|
| 91 |
client = OpenAI(
|
| 92 |
api_key="",
|
| 93 |
-
base_url=
|
| 94 |
http_client=httpx.Client(
|
| 95 |
event_hooks={
|
| 96 |
"request": [partial(proxy, model_info=MODELS[model_name])],
|
|
|
|
| 6 |
|
| 7 |
import gradio as gr
|
| 8 |
import httpx
|
| 9 |
+
from const import CLI_COMMAND, CSS, FOOTER, HEADER, MODELS, PLACEHOLDER
|
| 10 |
from openai import OpenAI
|
| 11 |
from PIL import Image
|
| 12 |
|
|
|
|
| 90 |
|
| 91 |
client = OpenAI(
|
| 92 |
api_key="",
|
| 93 |
+
base_url=MODELS[model_name]['host'],
|
| 94 |
http_client=httpx.Client(
|
| 95 |
event_hooks={
|
| 96 |
"request": [partial(proxy, model_info=MODELS[model_name])],
|
const.py
CHANGED
|
@@ -9,11 +9,10 @@ os.environ[SECRET_PREFIX + "PROJECT_ID"] = PROJECT_ID
|
|
| 9 |
os.environ[SECRET_PREFIX + "ROLE_SUBJECT"] = ROLE_SUBJECT
|
| 10 |
os.environ[SECRET_PREFIX + "CREDENTIALS"] = CREDENTIALS
|
| 11 |
|
| 12 |
-
BASE_URL = os.environ.get("GCLOUD_BASE_URL", "")
|
| 13 |
-
BASE_ENDPOINT = os.environ.get("GCLOUD_ENDPOINT", "")
|
| 14 |
-
CHATTY_ENDPOINT = os.environ.get("GCLOUD_CHATTY_ENDPOINT", "")
|
| 15 |
BASE_HOST = os.environ.get("GCLOUD_HOST", "")
|
|
|
|
| 16 |
CHATTY_HOST = os.environ.get("GCLOUD_CHATTY_HOST", "")
|
|
|
|
| 17 |
CLI_COMMAND_NAME = os.environ.get("CLI_COMMAND_NAME", "")
|
| 18 |
CLI_ARG1 = os.environ.get("CLI_ARG1", "")
|
| 19 |
CLI_ARG2 = os.environ.get("CLI_ARG2", "")
|
|
@@ -22,8 +21,8 @@ GCLOUD_BIN = str(ROOT_DIR / CLI_COMMAND_NAME)
|
|
| 22 |
CLI_COMMAND = [GCLOUD_BIN, CLI_ARG1, CLI_ARG2]
|
| 23 |
|
| 24 |
MODELS = {
|
| 25 |
-
"llava-calm2-siglip-chatty": {"
|
| 26 |
-
"llava-calm2-siglip": {"
|
| 27 |
}
|
| 28 |
|
| 29 |
HEADER = """
|
|
|
|
| 9 |
os.environ[SECRET_PREFIX + "ROLE_SUBJECT"] = ROLE_SUBJECT
|
| 10 |
os.environ[SECRET_PREFIX + "CREDENTIALS"] = CREDENTIALS
|
| 11 |
|
|
|
|
|
|
|
|
|
|
| 12 |
BASE_HOST = os.environ.get("GCLOUD_HOST", "")
|
| 13 |
+
BASE_ENDPOINT = os.environ.get("GCLOUD_ENDPOINT", "")
|
| 14 |
CHATTY_HOST = os.environ.get("GCLOUD_CHATTY_HOST", "")
|
| 15 |
+
CHATTY_ENDPOINT = os.environ.get("GCLOUD_CHATTY_ENDPOINT", "")
|
| 16 |
CLI_COMMAND_NAME = os.environ.get("CLI_COMMAND_NAME", "")
|
| 17 |
CLI_ARG1 = os.environ.get("CLI_ARG1", "")
|
| 18 |
CLI_ARG2 = os.environ.get("CLI_ARG2", "")
|
|
|
|
| 21 |
CLI_COMMAND = [GCLOUD_BIN, CLI_ARG1, CLI_ARG2]
|
| 22 |
|
| 23 |
MODELS = {
|
| 24 |
+
"llava-calm2-siglip-chatty": {"host": CHATTY_HOST, "endpoint": CHATTY_ENDPOINT},
|
| 25 |
+
"llava-calm2-siglip": {"host": BASE_HOST, "endpoint": BASE_ENDPOINT},
|
| 26 |
}
|
| 27 |
|
| 28 |
HEADER = """
|