bgregorutti commited on
Commit
bbfa082
·
1 Parent(s): 50281b0

Update app.py

Browse files

just a comment

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ if text:
43
 
44
 
45
  conn.close()
46
- generator = pipeline("text-generation", model="gpt2")
47
  txt=generator(
48
  text,
49
  max_length=150,
 
43
 
44
 
45
  conn.close()
46
+ generator = pipeline("text-generation", model="gpt2") # to modify for another model
47
  txt=generator(
48
  text,
49
  max_length=150,