api_for_unity / app.py
ldhldh's picture
Update app.py
c7a4ce6
raw
history blame
1.7 kB
from threading import Thread
import gradio as gr
import inspect
from gradio import routes
from typing import List, Type
import requests, os, re, asyncio, queue
import math
import time
import datetime
import requests, json
loop = asyncio.get_event_loop()
# 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
q = queue.Queue()
arrr = []
from huggingface_hub import hf_hub_download
hf_hub_download(repo_id='StarFox7/Llama-2-ko-7B-chat-gguf', filename='Llama-2-ko-7B-chat-gguf-q4_0.bin', local_dir='./')
# App code
def res(x, id, cdata, url):
global q
arr = [x, id, str(cdata.split(",", 1)[0]), url]
q.put(arr)
arrr.append(x)
print("\n_Done\n\n")
return "Done"
def rese(x):
print(f"{arrr}")
return "Done"
with gr.Blocks() as demo:
count = 0
aa = gr.Interface(
fn=res,
inputs=["text","text", "text", "text"],
outputs="text",
description="call",
)
bb = gr.Interface(
fn=rese,
inputs=["text"],
outputs="text",
description="call",
)
demo.queue(max_size=32).launch(enable_queue=True)