RashiAgarwal commited on
Commit
b9d49cf
1 Parent(s): d497b84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ model.load_adapter(adapter_path)
22
 
23
  def inference(prompt):
24
 
25
- pipe = pipeline(task="text-generation",model=model,tokenizer=tokenizer,max_length=200)
26
  result = pipe(f"<s>[INST] {prompt} [/INST]")
27
  return result[0]['generated_text']
28
 
 
22
 
23
  def inference(prompt):
24
 
25
+ pipe = pipeline(task="text-generation",model=model,tokenizer=tokenizer,max_length = 100)
26
  result = pipe(f"<s>[INST] {prompt} [/INST]")
27
  return result[0]['generated_text']
28