Update app.py

app.py

@@ -20,8 +20,9 @@ def translate(text):
     prompt = f"<s>[INST] translate this sentence to sinhala - {text} [/INST] sure,here the translation of the provided text - "
     input_ids = tokenizer.encode(prompt, return_tensors='pt')
     output = model.generate(input_ids, max_length=250, num_return_sequences=1)
-    for i, generated in enumerate(output):
-        translation = tokenizer.decode(generated, skip_special_tokens=True)
+    print(output[0])
+    translation = tokenizer.decode(output[0], skip_special_tokens=True)
+    print(translation)
     return translation