Spaces:
Sleeping
Sleeping
Commit
·
efa9e89
1
Parent(s):
d55c571
Update Text-Generation.py
Browse files- Text-Generation.py +0 -5
Text-Generation.py
CHANGED
|
@@ -73,11 +73,6 @@ def load_the_model():
|
|
| 73 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 74 |
return model
|
| 75 |
|
| 76 |
-
"""
|
| 77 |
-
with st.spinner("Please wait while the model loads:"):
|
| 78 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 79 |
-
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 80 |
-
"""
|
| 81 |
|
| 82 |
model = load_the_model()
|
| 83 |
tokenizer = load_the_tokenizer()
|
|
|
|
| 73 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
| 74 |
return model
|
| 75 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 76 |
|
| 77 |
model = load_the_model()
|
| 78 |
tokenizer = load_the_tokenizer()
|