Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ def generate_text(model_name, text, num_beams, max_length, top_p, temperature, r
|
|
8 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
9 |
|
10 |
# Initialize pipeline
|
11 |
-
pipe = pipeline("text-generation", model=
|
12 |
|
13 |
# Generate text with the specified parameters
|
14 |
generated_text = pipe(text,
|
|
|
8 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
9 |
|
10 |
# Initialize pipeline
|
11 |
+
pipe = pipeline("text-generation", model=model_name)
|
12 |
|
13 |
# Generate text with the specified parameters
|
14 |
generated_text = pipe(text,
|