Update app.py

app.py

@@ -20,7 +20,7 @@ def predict_identifier(tokenizer,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,max_new_tokens=250)[0]
+    output = model.generate(**input)[0]
     return tokenizer.decode(output)