Update app.py

app.py

@@ -8,14 +8,14 @@ tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
 
 
 
-prompt=''' Review the Following text as a human, who is tasked to Extract `Code Snippets` ,if any. If there are no Code Snippets in the below Text then return No CODE HIDDEN:
 
-"{}"
-'''
 def reply(message, history):
     m=message.replace('\n','')
     m=m.replace('\t','')
-    inputs = tokenizer(prompt.format(m), return_tensors="pt")
+    inputs = tokenizer(f''' Review the Following text as a human, who is tasked to Extract `Code Snippets` ,if any. If there are no Code Snippets in the below Text then return No CODE HIDDEN:
+
+"{message}"
+''', return_tensors="pt")
     outputs = model.generate(**inputs)
     code=tokenizer.batch_decode(outputs, skip_special_tokens=True)
     return code