Update app.py
Browse files
app.py
CHANGED
@@ -1,8 +1,8 @@
|
|
1 |
import streamlit as st
|
2 |
-
from transformers import
|
3 |
|
4 |
# TO-DD: ??? λΆλΆμ μ½λλ₯Ό μμ±νμμ€
|
5 |
-
|
6 |
|
7 |
device = "cuda" # the device to load the model onto
|
8 |
|
|
|
1 |
import streamlit as st
|
2 |
+
from transformers import AutoModelForCausalLM
|
3 |
|
4 |
# TO-DD: ??? λΆλΆμ μ½λλ₯Ό μμ±νμμ€
|
5 |
+
AutoModelForCausalLM = AutoModelForCausalLM.from_pretrained(task="translation", model="maywell/Synatra-7B-v0.3-Translation", tokenizer="maywell/Synatra-7B-v0.3-Translation")
|
6 |
|
7 |
device = "cuda" # the device to load the model onto
|
8 |
|