Update app.py

app.py

@@ -22,3 +22,4 @@ output = model.generate(input_ids, attention_mask=attention_mask, max_length=100
 # Decode and print the generated code
 generated_code = tokenizer.decode(output[0], skip_special_tokens=True)
 print(generated_code)
+