Spaces:
Build error
Build error
disable caching
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import torch
|
|
| 5 |
import json
|
| 6 |
|
| 7 |
|
| 8 |
-
|
| 9 |
def load_pipeline(model_ckpt, device):
|
| 10 |
pipe = pipeline('text-generation', model=model_ckpt, device=device)
|
| 11 |
return pipe
|
|
|
|
| 5 |
import json
|
| 6 |
|
| 7 |
|
| 8 |
+
#@st.cache(allow_output_mutation=True)
|
| 9 |
def load_pipeline(model_ckpt, device):
|
| 10 |
pipe = pipeline('text-generation', model=model_ckpt, device=device)
|
| 11 |
return pipe
|