ldhldh commited on
Commit
3d5c996
Β·
1 Parent(s): 4287daf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -19
app.py CHANGED
@@ -8,7 +8,7 @@ import requests, os, re, asyncio
8
 
9
 
10
  loop = asyncio.get_event_loop()
11
- gradio_client = GrClient('https://ldhldh-demo.hf.space/')
12
  # Monkey patch
13
  def get_types(cls_set: List[Type], component: str):
14
  docset = []
@@ -29,28 +29,15 @@ def get_types(cls_set: List[Type], component: str):
29
  routes.get_types = get_types
30
 
31
  # App code
32
- def chat(x):
33
- x = f"[***λ„ˆλŠ” Assistantμž…λ‹ˆλ‹€. μƒλŒ€μ—κ²Œ λ‹€μ–‘ν•œ μ§ˆλ¬Έμ„ ν•˜λ©° λŒ€ν™”λ₯Ό 이끌고 μžˆμŠ΅λ‹ˆλ‹€. Humanμ—κ²Œ 긍정적이고, κ³΅κ°ν•˜λ©°, μ΅œλŒ€ν•œ 길게 λŒ€λ‹΅ν•΄μ£Όμ„Έμš”***]\nHuman: {x}\n\n###\nAssistant:"
34
-
35
- print("\n___________________\n" + f"{x}")
36
-
37
  result = gradio_client.predict(
38
  x,
39
  # str representing input in 'User input' Textbox component
40
- 0.91, # float, representing input in 'Top-p (nucleus sampling)' Slider component
41
- 10, # int, representing input in 'Top-k (nucleus sampling)' Slider component
42
- 0.65, # float, representing input in 'Temperature' Slider component
43
- 20, # int, representing input in 'Max New Tokens' Slider component
44
- 1.2, # float, representing input in 'repetition_penalty' Slider component
45
  fn_index=0
46
  )
47
  result = str(result)
48
- output = result[len(x.rsplit(':', 1)[0])+2:]
49
- output = re.sub('ν•˜ν•˜','γ…Žγ…Ž', output)
50
- output = output.split('띓')[0]
51
- output = output.split('endoftext')[0]
52
- output = re.sub('[=+#/\:@*\"β€»γ†γ€β€˜|\\\<\>\(\)\[\]`\'…》\”\β€œ\’·]', '', output)
53
- #output = re.sub('[a-zA-Z]',' ',output)
54
 
55
  return output
56
 
@@ -61,8 +48,7 @@ with gr.Blocks() as demo:
61
  inputs="text",
62
  outputs="text",
63
  description="chat",
64
- #examples= [[f"\nfriend: λ„ˆλŠ” 꿈이 뭐야? \nyou: "],[f"\nyou: λ„ˆλŠ” 무슨 색을 κ°€μž₯ μ’‹μ•„ν•΄? \nfriend: κΈ€μŽ„ λ„ˆλŠ”? \nyou: "]]
65
- examples= [[f"λ„ˆλŠ” 꿈이 뭐야?"],[f"λ„ˆλŠ” 무슨 색을 κ°€μž₯ μ’‹μ•„ν•΄?"]]
66
  )
67
 
68
  demo.queue(max_size=32).launch(enable_queue=True)
 
8
 
9
 
10
  loop = asyncio.get_event_loop()
11
+ gradio_client = GrClient('https://ldhldh-demoforkakaoapi.hf.space/')
12
  # Monkey patch
13
  def get_types(cls_set: List[Type], component: str):
14
  docset = []
 
29
  routes.get_types = get_types
30
 
31
  # App code
32
+ def chat(x):
 
 
 
 
33
  result = gradio_client.predict(
34
  x,
35
  # str representing input in 'User input' Textbox component
36
+ 5,
 
 
 
 
37
  fn_index=0
38
  )
39
  result = str(result)
40
+ output = result[len(x.rsplit(':', 1)[0]):]
 
 
 
 
 
41
 
42
  return output
43
 
 
48
  inputs="text",
49
  outputs="text",
50
  description="chat",
51
+ examples= [[f"λ„ˆλŠ” 꿈이 뭐야?"],[f"λ„ˆλŠ” 무슨 색을 κ°€μž₯ μ’‹μ•„ν•΄?"]]
 
52
  )
53
 
54
  demo.queue(max_size=32).launch(enable_queue=True)