minor change

app.py

@@ -40,11 +40,12 @@ def generate_response(input,history: list[tuple[str, str]],max_tokens, temperatu
         messages.extend([{"role": "user", "content": val.get("content")}, {"role": "assistant", "content": val.get("content")}]) if val else None
 
     messages.append({"role": "user", "content": input})
-    print(f'Time to generate the story: {time_story}')
+    
     start = time.time()
     output = generate_text(input,max_new_tokens=max_tokens, top_p=top_p, temperature=temperature)
     end = time.time()
     time_story= end-start
+    print(f'Time to generate the story: {time_story}')
     history.append((input,output))
     yield output