Update app.py
Browse files
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 = "
|
| 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)
|