Hugging Face's logo

Spaces:
csuvikv
/
gpt-3-secret
Runtime error

App Files Community
3v324v23 commited on
Commit
8ca6a19
·
1 Parent(s): 6d5c1f5

model switch

Browse files
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-6b', 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-2.7b', return_full_text=False, do_sample=False)
8
 
9
 
10
  def Bemenet(input_string):