Dhahlan2000 commited on
Commit
12e3736
·
verified ·
1 Parent(s): 5fd968a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "HuggingFaceH4/zephyr-7b-gemma-v0.1" # 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
 
 
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