Spaces:

Sasidhar
/

ml-playground

Runtime error

Sasidhar commited on Aug 18, 2022

Commit

07c1b54

1 Parent(s): 215d64a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,6 @@ def load_model(model_name):
     model = AutoModelWithLMHead.from_pretrained("gpt2-large")
     return model
-@st.cache
 def load_text_gen_model():
     generator = pipeline("text-generation", model="gpt2-medium")
     return generator

     model = AutoModelWithLMHead.from_pretrained("gpt2-large")
     return model
 def load_text_gen_model():
     generator = pipeline("text-generation", model="gpt2-medium")
     return generator