stt_api / app.py
ldhldh's picture
Update app.py
63c8167
raw
history blame
1.71 kB
import gradio as gr
from gradio_client import Client as GrClient
import inspect
from gradio import routes
from typing import List, Type
import requests, os, re, asyncio
loop = asyncio.get_event_loop()
gradio_client = GrClient('https://ldhldh-demo.hf.space/')
# Monkey patch
def get_types(cls_set: List[Type], component: str):
docset = []
types = []
if component == "input":
for cls in cls_set:
doc = inspect.getdoc(cls)
doc_lines = doc.split("\n")
docset.append(doc_lines[1].split(":")[-1])
types.append(doc_lines[1].split(")")[0].split("(")[-1])
else:
for cls in cls_set:
doc = inspect.getdoc(cls)
doc_lines = doc.split("\n")
docset.append(doc_lines[-1].split(":")[-1])
types.append(doc_lines[-1].split(")")[0].split("(")[-1])
return docset, types
routes.get_types = get_types
# App code
def chat(x):
x = f"[***๋„ˆ๋Š” Assistant์ž…๋‹ˆ๋‹ค. ์ƒ๋Œ€์—๊ฒŒ ๋‹ค์–‘ํ•œ ์งˆ๋ฌธ์„ ํ•˜๋ฉฐ ๋Œ€ํ™”๋ฅผ ์ด๋Œ๊ณ  ์žˆ์Šต๋‹ˆ๋‹ค. Human์—๊ฒŒ ๊ธ์ •์ ์ด๊ณ , ๊ณต๊ฐํ•˜๋ฉฐ, ์ตœ๋Œ€ํ•œ ๊ธธ๊ฒŒ ๋Œ€๋‹ตํ•ด์ฃผ์„ธ์š”***]\nHuman: {x}\n\n###\nAssistant:"
print("\n___________________\n" + f"{x}")
output = "Hello?"
return output
with gr.Blocks() as demo:
count = 0
aa = gr.Interface(
fn=chat,
inputs="text",
outputs="text",
description="chat",
#examples= [[f"\nfriend: ๋„ˆ๋Š” ๊ฟˆ์ด ๋ญ์•ผ? \nyou: "],[f"\nyou: ๋„ˆ๋Š” ๋ฌด์Šจ ์ƒ‰์„ ๊ฐ€์žฅ ์ข‹์•„ํ•ด? \nfriend: ๊ธ€์Ž„ ๋„ˆ๋Š”? \nyou: "]]
examples= [[f"๋„ˆ๋Š” ๊ฟˆ์ด ๋ญ์•ผ?"],[f"๋„ˆ๋Š” ๋ฌด์Šจ ์ƒ‰์„ ๊ฐ€์žฅ ์ข‹์•„ํ•ด?"]]
)
demo.queue(max_size=32).launch(enable_queue=True)