csuvikv commited on
Commit
df867ef
1 Parent(s): 5f6bbc2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline, set_seed, AutoTokenizer
4
 
5
  set_seed(42)
6
 
7
- generator = pipeline('text-generation', model='PygmalionAI/pygmalion-2.7b', return_full_text=False, do_sample=False)
8
 
9
 
10
  def Bemenet(input_string):
 
4
 
5
  set_seed(42)
6
 
7
+ generator = pipeline('text-generation', model='PygmalionAI/pygmalion-1.3b', return_full_text=False, do_sample=False)
8
 
9
 
10
  def Bemenet(input_string):