addp
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 2 |
import time
|
| 3 |
import datetime
|
| 4 |
|
|
@@ -21,10 +21,17 @@ messages = [
|
|
| 21 |
]
|
| 22 |
|
| 23 |
print('gerando a saida...')
|
|
|
|
|
|
|
|
|
|
| 24 |
pipe = pipeline("text-generation", model="01-ai/Yi-1.5-34B-Chat")
|
| 25 |
|
|
|
|
|
|
|
| 26 |
output = pipe(messages)
|
| 27 |
|
|
|
|
|
|
|
| 28 |
st.write(output)
|
| 29 |
|
| 30 |
|
|
|
|
| 1 |
+
# from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 2 |
import time
|
| 3 |
import datetime
|
| 4 |
|
|
|
|
| 21 |
]
|
| 22 |
|
| 23 |
print('gerando a saida...')
|
| 24 |
+
|
| 25 |
+
st.write('gerando a saida...')
|
| 26 |
+
|
| 27 |
pipe = pipeline("text-generation", model="01-ai/Yi-1.5-34B-Chat")
|
| 28 |
|
| 29 |
+
st.write('pipeline...')
|
| 30 |
+
|
| 31 |
output = pipe(messages)
|
| 32 |
|
| 33 |
+
st.write('saída gerada...')
|
| 34 |
+
|
| 35 |
st.write(output)
|
| 36 |
|
| 37 |
|