cloghost commited on
Commit
5717f8d
·
verified ·
1 Parent(s): a95ccf0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -45,8 +45,7 @@ model_name = "cloghost/nllb-200-distilled-600M-hin-kang-v1"
45
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
46
  tokenizer = AutoTokenizer.from_pretrained(model_name)
47
 
48
- device = 0 if torch.cuda.is_available() else -1 # 0 for GPU, -1 for CPU
49
-
50
  translator = pipeline(
51
  "translation",
52
  model=model,
@@ -59,6 +58,6 @@ translator = pipeline(
59
  text = """मगर हिमाचली भाषा तो पहले से बोली जा रही है।
60
  लोग सदियों से ही इसके संग जी रहे हैं।
61
  पहाड़ी भाषा का इतिहास हिन्दी साहित्य के आदिकाल ,‌जिसे सिद्ध चारण काल के नाम से भी जानते हैं
62
- """ # Example text in Hindi
63
 
64
  translation = translator(text)
 
45
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
46
  tokenizer = AutoTokenizer.from_pretrained(model_name)
47
 
48
+ device = 0 if torch.cuda.is_available() else -1
 
49
  translator = pipeline(
50
  "translation",
51
  model=model,
 
58
  text = """मगर हिमाचली भाषा तो पहले से बोली जा रही है।
59
  लोग सदियों से ही इसके संग जी रहे हैं।
60
  पहाड़ी भाषा का इतिहास हिन्दी साहित्य के आदिकाल ,‌जिसे सिद्ध चारण काल के नाम से भी जानते हैं
61
+ """
62
 
63
  translation = translator(text)