tyang4 commited on
Commit
c48e809
·
verified ·
1 Parent(s): 30be17c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -76,8 +76,8 @@ def gpt_chat(sys_msg: str, user_msg: str, **kwargs):
76
  class SimpleEncoder:
77
  def __init__(self):
78
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
79
- self.tokenizer = AutoTokenizer.from_pretrained("model")
80
- self.model = AutoModel.from_pretrained("model").to(self.device)
81
  self.model.eval()
82
 
83
  def encode(self, texts, batch_size: int = 16):
 
76
  class SimpleEncoder:
77
  def __init__(self):
78
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
79
+ self.tokenizer = AutoTokenizer.from_pretrained("/app/model")
80
+ self.model = AutoModel.from_pretrained("/app/model").to(self.device)
81
  self.model.eval()
82
 
83
  def encode(self, texts, batch_size: int = 16):