Update README.md

README.md

@@ -48,7 +48,7 @@ context = "Mataró és una ciutat de Catalunya, capital de la comarca del Maresm
 
 # We need to format the prompt and context using ### and \n
 
-def givePrediction(instruction, context, max_new_tokens=50, repetition_penalty=1.2, top_k=50, top_p=0.95, do_sample=True, temperature=0.5)
+def givePrediction(instruction, context, max_new_tokens=50, repetition_penalty=1.2, top_k=50, top_p=0.95, do_sample=True, temperature=0.5):
     text = f"### Instruction\n{{instruction}}\n### Context\n{{context}}\n### Answer\n"
     response = pipe(text.format(instruction=instruction, context=context),temperature=temperature,repetition_penalty=repetition_penalty, max_new_tokens=max_new_tokens,top_k=top_k, top_p=top_p, do_sample=do_sample)[0]["generated_text"]
     answer = response.split("###")[-1][8:-1]