Update app.py

app.py

@@ -10,9 +10,9 @@ def generate_code(NL):
     inputs = tokenizer([NL], padding="max_length", truncation=True, max_length=512, return_tensors="pt")
     input_ids = inputs.input_ids
     attention_mask = inputs.attention_mask
-    outputs = new_model.generate(input_ids, attention_mask=attention_mask)
-    # all special tokens including will be removed
-    output_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    outputs = model.generate(input_ids, attention_mask=attention_mask)
+
+    output_code = tokenizer.decode(outputs, skip_special_tokens=True)
     return output_code