devendergarg14 commited on
Commit
31dbe0a
1 Parent(s): 11626ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from transformers import pipeline
2
- generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
3
  def query(input_sentence,num,start):
4
  string3=[]
5
  for i in range(0,num):
 
1
  from transformers import pipeline
2
+ generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
3
  def query(input_sentence,num,start):
4
  string3=[]
5
  for i in range(0,num):