project-baize commited on
Commit
137041a
1 Parent(s): e0ec530

Improve UI

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -26,7 +26,7 @@ def predict(text,
26
  max_length_tokens,
27
  max_context_length_tokens,):
28
  if text=="":
29
- yield chatbot,history,"Empty Context"
30
  return
31
  try:
32
  model
@@ -36,7 +36,7 @@ def predict(text,
36
 
37
  inputs = generate_prompt_with_history(text,history,tokenizer,max_length=max_context_length_tokens)
38
  if inputs is None:
39
- yield chatbot,history,"Too Long Input"
40
  return
41
  else:
42
  prompt,inputs=inputs
@@ -55,11 +55,11 @@ def predict(text,
55
  x = x[:x.index("[|AI|]")].strip()
56
  x = x.strip(" ")
57
  a, b= [[y[0],convert_to_markdown(y[1])] for y in history]+[[text, convert_to_markdown(x)]],history + [[text,x]]
58
- yield a, b, "Generating……"
59
  if shared_state.interrupted:
60
  shared_state.recover()
61
  try:
62
- yield a, b, "Stop Success"
63
  return
64
  except:
65
  pass
@@ -68,7 +68,7 @@ def predict(text,
68
  #print(x)
69
  #print("="*80)
70
  try:
71
- yield a,b,"Generate Success"
72
  except:
73
  pass
74
 
@@ -81,7 +81,7 @@ def retry(
81
  max_length_tokens,
82
  max_context_length_tokens,
83
  ):
84
- logging.info("Retry……")
85
  if len(history) == 0:
86
  yield chatbot, history, f"Empty context"
87
  return
 
26
  max_length_tokens,
27
  max_context_length_tokens,):
28
  if text=="":
29
+ yield chatbot,history,"Empty context."
30
  return
31
  try:
32
  model
 
36
 
37
  inputs = generate_prompt_with_history(text,history,tokenizer,max_length=max_context_length_tokens)
38
  if inputs is None:
39
+ yield chatbot,history,"Input too long."
40
  return
41
  else:
42
  prompt,inputs=inputs
 
55
  x = x[:x.index("[|AI|]")].strip()
56
  x = x.strip(" ")
57
  a, b= [[y[0],convert_to_markdown(y[1])] for y in history]+[[text, convert_to_markdown(x)]],history + [[text,x]]
58
+ yield a, b, "Generating..."
59
  if shared_state.interrupted:
60
  shared_state.recover()
61
  try:
62
+ yield a, b, "Stop: Success"
63
  return
64
  except:
65
  pass
 
68
  #print(x)
69
  #print("="*80)
70
  try:
71
+ yield a,b,"Generate: Success"
72
  except:
73
  pass
74
 
 
81
  max_length_tokens,
82
  max_context_length_tokens,
83
  ):
84
+ logging.info("Retry...")
85
  if len(history) == 0:
86
  yield chatbot, history, f"Empty context"
87
  return