Flexy0 commited on
Commit
0516f56
β€’
1 Parent(s): 2c3a42e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ model.save_pretrained("./gpt4-text-gen")
36
  tokenizer.save_pretrained("./gpt4-text-gen")
37
 
38
  # Generate text using the fine-tuned model
39
- input_text = "Once upon a time"
40
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
41
  output = model.generate(input_ids, max_length=100, num_return_sequences=1, no_repeat_ngram_size=2, top_k=50, top_p=0.95)
42
  generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
 
36
  tokenizer.save_pretrained("./gpt4-text-gen")
37
 
38
  # Generate text using the fine-tuned model
39
+ input_text = "Hello!"
40
  input_ids = tokenizer.encode(input_text, return_tensors="pt")
41
  output = model.generate(input_ids, max_length=100, num_return_sequences=1, no_repeat_ngram_size=2, top_k=50, top_p=0.95)
42
  generated_text = tokenizer.decode(output[0], skip_special_tokens=True)