daresearch commited on
Commit
1fdf5d1
·
verified ·
1 Parent(s): 55a510c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ model_with_adapter = PeftModel.from_pretrained(
27
  tokenizer = AutoTokenizer.from_pretrained(base_model_name)
28
 
29
  # Step 5: Define the inference function
30
- def generate_text(prompt, max_length=100):
31
  # Tokenize the input
32
  inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
33
 
 
27
  tokenizer = AutoTokenizer.from_pretrained(base_model_name)
28
 
29
  # Step 5: Define the inference function
30
+ def generate_text(prompt, max_length=1024):
31
  # Tokenize the input
32
  inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
33