Icyud commited on
Commit
382dc10
1 Parent(s): a7c8991

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import streamlit as st
2
- from stranformer import pipeline
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)