NV9523 commited on
Commit
a048055
·
verified ·
1 Parent(s): ea74705

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import subprocess
2
  import sys
3
 
4
- # Cài đặt thư viện nếu chưa có
5
- subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers", "streamlit", "torch", "peft"])
6
 
7
  import streamlit as st
8
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
 
1
  import subprocess
2
  import sys
3
 
4
+ # Cài đặt các thư viện nếu chưa có
5
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers", "streamlit", "torch", "bitsandbytes"])
6
 
7
  import streamlit as st
8
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer