Spaces:
Sleeping
Sleeping
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 |
|