Spaces:

Komal-patra
/

EU_AI_ACT

Sleeping

Komal-patra commited on Jul 7, 2024

Commit

f95e467

verified ·

1 Parent(s): 69b124f

updated

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ base_model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 model = PeftModel.from_pretrained(base_model, "Komal-patra/results")
 # Text generation function
-def generate_text(prompt, max_length=150):
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         input_ids=inputs["input_ids"],

 model = PeftModel.from_pretrained(base_model, "Komal-patra/results")
 # Text generation function
+def generate_text(prompt, max_length=512):
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         input_ids=inputs["input_ids"],