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)
|