Update app.py

app.py

@@ -26,9 +26,7 @@ def generate(prompt, history, max_new_tokens = 128, temperature = 0.6):
     formatted_prompt = format_prompt(prompt, history)
     response = ""
 
-    # Count the number of tokens in the prompt
     num_prompt_tokens = len(tokenizer(formatted_prompt)['input_ids'])
-    # Calculate the maximum length for the generation
     max_length = num_prompt_tokens + max_new_tokens
 
     textgen = pipeline('text-generation', model=model, tokenizer=tokenizer, max_length=max_length, temperature=temperature)
@@ -36,11 +34,9 @@ def generate(prompt, history, max_new_tokens = 128, temperature = 0.6):
     response = output[0]['generated_text'].replace(formatted_prompt, '')
     return response
 
-
 mychatbot = gr.Chatbot(
     avatar_images=["user.png", "botp.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True,)
 
-
 demo  = gr.ChatInterface(fn=generate,
                         chatbot=mychatbot,
                         title="Phi-3 Mini Chat Demo",