Spaces:
Sleeping
Sleeping
Dhahlan2000
commited on
Commit
•
2631383
1
Parent(s):
a413ffd
Update app.py
Browse files
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 |
|