MJ106 commited on
Commit
798e894
ยท
verified ยท
1 Parent(s): 293f78e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import streamlit as st
2
- from transformers import pipeline
3
 
4
  # TO-DD: ??? ๋ถ€๋ถ„์˜ ์ฝ”๋“œ๋ฅผ ์™„์„ฑํ•˜์‹œ์˜ค
5
- pipeline = 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
 
 
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