Chris STC
commited on
Commit
·
d233d23
1
Parent(s):
e7c4ff2
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ with gr.Blocks(theme=theme) as demo:
|
|
43 |
history[-1][1] = ""
|
44 |
count = 0
|
45 |
output = ""
|
46 |
-
for token in llm2.generate(
|
47 |
text = llm2.detokenize([token])
|
48 |
output += text.decode()
|
49 |
count += 1
|
@@ -59,4 +59,4 @@ with gr.Blocks(theme=theme) as demo:
|
|
59 |
gr.HTML("Thanks for checking out this app!")
|
60 |
|
61 |
demo.queue()
|
62 |
-
demo.launch(debug=True)
|
|
|
43 |
history[-1][1] = ""
|
44 |
count = 0
|
45 |
output = ""
|
46 |
+
for token in llm2.generate(tokens): # (tokens, top_k=50, top_p=0.73, temp=0.72, repeat_penalty=1.1):
|
47 |
text = llm2.detokenize([token])
|
48 |
output += text.decode()
|
49 |
count += 1
|
|
|
59 |
gr.HTML("Thanks for checking out this app!")
|
60 |
|
61 |
demo.queue()
|
62 |
+
demo.launch(debug=True, share=False)
|