rajj0 commited on
Commit
0be83ad
·
1 Parent(s): cb5391c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from peft import PeftModel
7
  model1 = "rajj0/abstract_ai"
8
  tokenizer = "NousResearch/llama-2-7b-chat-hf"
9
  #model = PeftModel.from_pretrained(model1)
10
- pipe = pipeline(task="text-generation", model=model1, tokenizer=tokenizer, max_length=1000, temperature=0.7, top_p=0.92)
11
  def summarize(text):
12
  input = pipe(text)
13
  return input
 
7
  model1 = "rajj0/abstract_ai"
8
  tokenizer = "NousResearch/llama-2-7b-chat-hf"
9
  #model = PeftModel.from_pretrained(model1)
10
+ pipe = pipeline(task="text-generation", model=model1, tokenizer=tokenizer, max_length=100, temperature=0.7, top_p=0.92)
11
  def summarize(text):
12
  input = pipe(text)
13
  return input