rasyosef commited on
Commit
350fc62
·
verified ·
1 Parent(s): 34fe501

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ phi1_5 = pipeline(
22
  )
23
 
24
  # Function that accepts a prompt and generates text using the phi2 pipeline
25
- def generate(message, chat_history, max_new_tokens):
26
 
27
  history = [
28
  {"role": "system", "content": "You are Phi, a helpful AI assistant made by Microsoft and RasYosef. User will you give you a task. Your goal is to complete the task as faithfully as you can."}
 
22
  )
23
 
24
  # Function that accepts a prompt and generates text using the phi2 pipeline
25
+ def generate(message, chat_history, max_new_tokens=256):
26
 
27
  history = [
28
  {"role": "system", "content": "You are Phi, a helpful AI assistant made by Microsoft and RasYosef. User will you give you a task. Your goal is to complete the task as faithfully as you can."}