Dhahlan2000 commited on
Commit
e706b2f
·
verified ·
1 Parent(s): 6e3bcaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def transliterate_to_sinhala(text):
46
  # Load conversation model
47
  conv_model_name = "microsoft/Phi-3-mini-4k-instruct" # Use GPT-2 instead of the gated model
48
  tokenizer = AutoTokenizer.from_pretrained(conv_model_name, trust_remote_code=True)
49
- model = AutoModelForCausalLM.from_pretrained(conv_model_name).to(device)
50
 
51
  def conversation_predict(text):
52
  pipe = pipeline(
 
46
  # Load conversation model
47
  conv_model_name = "microsoft/Phi-3-mini-4k-instruct" # Use GPT-2 instead of the gated model
48
  tokenizer = AutoTokenizer.from_pretrained(conv_model_name, trust_remote_code=True)
49
+ model = AutoModelForCausalLM.from_pretrained(conv_model_name, `trust_remote_code=True`).to(device)
50
 
51
  def conversation_predict(text):
52
  pipe = pipeline(