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