Spaces:
Runtime error
Runtime error
Commit
•
d88ecb8
1
Parent(s):
7f30cff
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from sklearn.metrics.pairwise import cosine_similarity, manhattan_distances, euc
|
|
| 8 |
|
| 9 |
|
| 10 |
|
| 11 |
-
@st.cache
|
| 12 |
def load_model():
|
| 13 |
tokenizer = AutoTokenizer.from_pretrained("stanford-crfm/pubmedgpt")
|
| 14 |
model = AutoModel.from_pretrained("stanford-crfm/pubmedgpt")
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
|
| 11 |
+
@st.cache(allow_output_mutation=True)
|
| 12 |
def load_model():
|
| 13 |
tokenizer = AutoTokenizer.from_pretrained("stanford-crfm/pubmedgpt")
|
| 14 |
model = AutoModel.from_pretrained("stanford-crfm/pubmedgpt")
|