opt13b-gradio / main.py
adpai's picture
Create main.py
5035392
raw
history blame contribute delete
355 Bytes
from transformers import pipeline
# Initialize the text generation pipeline with the specified model
generator = pipeline('text-generation', model='KoboldAI/OPT-13B-Erebus')
# Generate text
generated_text = generator("Welcome Captain Janeway, I apologize for the delay.", do_sample=True, min_length=50)
# Print the generated text
print(generated_text)