MJ106 commited on
Commit
798e894
β€’
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