GadaisAI commited on
Commit
d90fb13
1 Parent(s): 001bcc1

Create Model.py

Browse files

Sorry for the simplified model.

Files changed (1) hide show
  1. Model.py +10 -0
Model.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import torch,random
3
+ from transformers import pipeline
4
+ fill_mask = pipeline("fill-mask")
5
+ while True:
6
+ x=input(">")
7
+ for i in range(42):
8
+ x+=" <mask>"
9
+ x=fill_mask(x)[0]["sequence"]
10
+ print(x)