mksaad commited on
Commit
8993d82
·
verified ·
1 Parent(s): b9633bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ import re
13
 
14
  # Use a pipeline as a high-level helper
15
  from transformers import pipeline
16
- pipe = pipeline("text-generation", model="openai-community/gpt2-medium")
 
17
 
18
  def get_clean_text(url):
19
  try:
 
13
 
14
  # Use a pipeline as a high-level helper
15
  from transformers import pipeline
16
+ # pipe = pipeline("text-generation", model="openai-community/gpt2-medium")
17
+ pipe = pipeline("text-generation", model="openai-community/gpt2")
18
 
19
  def get_clean_text(url):
20
  try: