Update app.py

app.py

@@ -13,11 +13,11 @@ def generate(text, max_length, num_return_sequences=1):
     inputs = tokenizer.encode(text, padding=False, add_special_tokens=False, return_tensors="pt")
     outputs = model.generate(inputs, max_length=max_length, num_return_sequences=num_return_sequences)
     gen_text = "Assignment : " + tokenizer.decode(outputs[0]).split("#")[0] if "#" else "Assignment : " + tokenizer.decode(outputs[0])
+
     return gen_text
 
 
-def extract_functions
-(text):
+def extract_functions(text):
     function_pattern = r'def\s+(\w+)\((.*?)\):([\s\S]*?)return\s+(.*?)\n'
     functions = re.findall(function_pattern, text, flags=re.MULTILINE)
     extracted_text = []