ldhldh commited on
Commit
2475fa5
·
1 Parent(s): f9b97c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -24
app.py CHANGED
@@ -1,30 +1,6 @@
1
- from threading import Thread
2
  import gradio as gr
3
- import inspect, asyncio
4
- from gradio import routes
5
- from typing import List, Type
6
  from transformers import AutoTokenizer
7
 
8
- loop = asyncio.get_event_loop()
9
-
10
- # init code
11
- def get_types(cls_set: List[Type], component: str):
12
- docset = []
13
- types = []
14
- if component == "input":
15
- for cls in cls_set:
16
- doc = inspect.getdoc(cls)
17
- doc_lines = doc.split("\n")
18
- docset.append(doc_lines[1].split(":")[-1])
19
- types.append(doc_lines[1].split(")")[0].split("(")[-1])
20
- else:
21
- for cls in cls_set:
22
- doc = inspect.getdoc(cls)
23
- doc_lines = doc.split("\n")
24
- docset.append(doc_lines[-1].split(":")[-1])
25
- types.append(doc_lines[-1].split(")")[0].split("(")[-1])
26
- return docset, types
27
- routes.get_types = get_types
28
 
29
  from petals import AutoDistributedModelForCausalLM
30
  import npc_data
 
 
1
  import gradio as gr
 
 
 
2
  from transformers import AutoTokenizer
3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
 
5
  from petals import AutoDistributedModelForCausalLM
6
  import npc_data