Update app.py

app.py

@@ -104,20 +104,20 @@ def process():
     
         # Generate text
         with torch.no_grad():
-            #outputs = model.generate(input_ids)
-            #generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            outputs = model.generate(input_ids)
+            generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
         #To Here
    
-            outputs = model.generate(input_ids,
-                                     min_length = 20,
-                                     max_new_tokens = 600,
-                                     length_penalty = 1.0, # Set to values < 1.0 in order to encourage the model to generate shorter answers.
-                                     num_beams = 10,
-                                     no_repeat_ngram_size = 3,
-                                     temperature = 0,
-                                     top_k = 150, # default 50
-                                     top_p = 0.92,
-                                     repetition_penalty = 2.1)
-            generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            #outputs = model.generate(input_ids,
+                                     #min_length = 20,
+                                     #max_new_tokens = 600,
+                                     #length_penalty = 1.0, # Set to values < 1.0 in order to encourage the model to generate shorter answers.
+                                     #num_beams = 10,
+                                     #no_repeat_ngram_size = 3,
+                                     #temperature = 0,
+                                     #top_k = 150, # default 50
+                                     #top_p = 0.92,
+                                     #repetition_penalty = 2.1)
+            #generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
         st.write(GoogleTranslator(source='en', target='hy').translate(generated_text))
 process()