Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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
|