Update app.py

app.py

@@ -14,7 +14,7 @@ def use_graphcodebert(input):
   # Encode input
   input_ids = tokenizer.encode(input, return_tensors="pt")
   # Generate output
-  output_ids = model.generate(input_ids, max_length=50)
+  output_ids = model.generate(input_ids, max_length=5000)
   # Decode output
   output = tokenizer.decode(output_ids[0], skip_special_tokens=True)
   # Return output