Update app.py

app.py

@@ -59,7 +59,7 @@ def predict_identifier(code):
     code should be like: "unsigned __int8 *__cdecl <func>(int *<var_0>,...){ return <func_1>(1);}"
     '''
     input = tokenizer('identifier_predict: '+code, return_tensors='pt', max_length=max_input_length, truncation=True)
-    output = model.generate(**input)[0]
+    output = model.generate(**input, max_new_tokens=10000)[0]
     return tokenizer.decode(output)
 
 # Create the Gradio interface for predicting identifiers