Update src/streamlit_app.py
Browse files- 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):
|