Dhahlan2000 commited on
Commit
2631383
1 Parent(s): a413ffd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -99,10 +99,11 @@ def transliterate_to_sinhala(text):
99
  latin_text = transliterate.process(source_script, target_script, text)
100
  return latin_text
101
 
102
- tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")
103
  model = AutoModelForCausalLM.from_pretrained(
104
  "google/gemma-2b-it",
105
- torch_dtype=torch.bfloat16
 
106
  )
107
 
108
 
 
99
  latin_text = transliterate.process(source_script, target_script, text)
100
  return latin_text
101
 
102
+ tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it", token = access_token)
103
  model = AutoModelForCausalLM.from_pretrained(
104
  "google/gemma-2b-it",
105
+ torch_dtype=torch.bfloat16,
106
+ token = access_token
107
  )
108
 
109