Spaces:
Running
Running
Dhahlan2000
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ def transliterate_to_sinhala(text):
|
|
44 |
return transliterate.process('Velthuis', 'Sinhala', text)
|
45 |
|
46 |
# Load conversation model
|
47 |
-
conv_model_name = "
|
48 |
tokenizer = AutoTokenizer.from_pretrained(conv_model_name)
|
49 |
model = AutoModelForCausalLM.from_pretrained(conv_model_name).to(device)
|
50 |
|
|
|
44 |
return transliterate.process('Velthuis', 'Sinhala', text)
|
45 |
|
46 |
# Load conversation model
|
47 |
+
conv_model_name = "Qwen/Qwen2-0.5B-Instruct" # Use GPT-2 instead of the gated model
|
48 |
tokenizer = AutoTokenizer.from_pretrained(conv_model_name)
|
49 |
model = AutoModelForCausalLM.from_pretrained(conv_model_name).to(device)
|
50 |
|