Update app.py

app.py

@@ -9,7 +9,7 @@ tokenizer = AutoTokenizer.from_pretrained(model)
 
 # predict summary
 def predict_summary(tokenizer,code):
-    input = tokenizer('summarize: '+code,return_tensors='pt',max_length=max_input_length,truncation=>
+    input = tokenizer('summarize: '+code,return_tensors='pt',max_length=max_input_length,truncation=True)
     output = model.generate(**input,max_new_tokens=256)[0]
     return tokenizer.decode(output,skip_special_tokens=True)
 
@@ -19,7 +19,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,tr>
+    input = tokenizer('identifier_predict: '+code,return_tensors='pt',max_length=max_input_length,truncation=True)
     output = model.generate(**input,max_new_tokens=250)[0]
     return tokenizer.decode(output)