peterkchung commited on
Commit
1d5b074
·
verified ·
1 Parent(s): ded0246

Update app.py conditional error fix

Browse files
Files changed (1) hide show
  1. app.py +26 -23
app.py CHANGED
@@ -69,30 +69,33 @@ def retry_query(
69
  lookback = 3,
70
  max_new_tokens = 256,
71
  ):
72
-
73
- query = history[-1][0]
74
- history[-1][1] = None
75
-
76
- generateKwargs = dict(
77
- max_new_tokens = max_new_tokens,
78
- seed = 1337,
79
- )
80
-
81
- formatted_query = format_prompt(query, history, lookback)
82
-
83
- stream = client.text_generation(
84
- formatted_query,
85
- **generateKwargs,
86
- stream = True,
87
- details = True,
88
- return_full_text = False
89
- )
90
-
91
- history[-1][1] = ""
92
 
93
- for response in stream:
94
- history[-1][1] += response.token.text
95
- yield history
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
96
 
97
 
98
  """
 
69
  lookback = 3,
70
  max_new_tokens = 256,
71
  ):
72
+ if history == None:
73
+ pass
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
74
 
75
+ else:
76
+ query = history[-1][0]
77
+ history[-1][1] = None
78
+
79
+ generateKwargs = dict(
80
+ max_new_tokens = max_new_tokens,
81
+ seed = 1337,
82
+ )
83
+
84
+ formatted_query = format_prompt(query, history, lookback)
85
+
86
+ stream = client.text_generation(
87
+ formatted_query,
88
+ **generateKwargs,
89
+ stream = True,
90
+ details = True,
91
+ return_full_text = False
92
+ )
93
+
94
+ history[-1][1] = ""
95
+
96
+ for response in stream:
97
+ history[-1][1] += response.token.text
98
+ yield history
99
 
100
 
101
  """