riotu-lab commited on
Commit
c208291
1 Parent(s): e792a4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=model, tokenizer=tokenizer)
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,