Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import streamlit as st
|
2 |
-
from
|
3 |
|
4 |
base_model = "minhtt/vistral-7b-chat"
|
5 |
|
@@ -32,7 +32,7 @@ tokenizer.add_eos_token = True
|
|
32 |
tokenizer.bos_token, tokenizer.eos_token
|
33 |
|
34 |
pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
|
35 |
-
text = st.text_erea()
|
36 |
|
37 |
if text:
|
38 |
out = pipe(text)
|
|
|
1 |
import streamlit as st
|
2 |
+
from transformer import pipeline
|
3 |
|
4 |
base_model = "minhtt/vistral-7b-chat"
|
5 |
|
|
|
32 |
tokenizer.bos_token, tokenizer.eos_token
|
33 |
|
34 |
pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
|
35 |
+
text = st.text_erea("Đặt câu hỏi")
|
36 |
|
37 |
if text:
|
38 |
out = pipe(text)
|