adpai commited on
Commit
5035392
·
1 Parent(s): b48d151

Create main.py

Browse files
Files changed (1) hide show
  1. main.py +10 -0
main.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ # Initialize the text generation pipeline with the specified model
4
+ generator = pipeline('text-generation', model='KoboldAI/OPT-13B-Erebus')
5
+
6
+ # Generate text
7
+ generated_text = generator("Welcome Captain Janeway, I apologize for the delay.", do_sample=True, min_length=50)
8
+
9
+ # Print the generated text
10
+ print(generated_text)