Update app.py

app.py

@@ -43,16 +43,18 @@ examples = [
 ]
 
 # predict summary
-def predict_summary(tokenizer,code):
+def predict_summary(code):
     global model
+    global tokenizer
     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)
 
 
 # predict identifier (func name)
-def predict_identifier(tokenizer,code):
+def predict_identifier(code):
     global model
+    global tokenizer
     '''
     code should be like: "unsigned __int8 *__cdecl <func>(int *<var_0>,...){ return <func_1>(1);}"
     '''