Update pages/GPT.py
Browse files- pages/GPT.py +1 -3
pages/GPT.py
CHANGED
|
@@ -4,8 +4,6 @@ import numpy as np
|
|
| 4 |
import transformers
|
| 5 |
import random
|
| 6 |
import textwrap
|
| 7 |
-
import warnings
|
| 8 |
-
warnings.filterwarnings('ignore')
|
| 9 |
|
| 10 |
@st.cache
|
| 11 |
def load_model():
|
|
@@ -47,7 +45,7 @@ max_len = st.slider('Length of sequence', 0, 500, 250)
|
|
| 47 |
temp = st.slider('Temperature', 1, 30, 1)
|
| 48 |
num_generate = st.text_input("Enter number of sequences")
|
| 49 |
|
| 50 |
-
if st.button('Generate')
|
| 51 |
prompt = preprocess_text(text_input, tokenizer)
|
| 52 |
result = predict_sentiment(model, prompt, temp, int(num_generate))
|
| 53 |
for i in result:
|
|
|
|
| 4 |
import transformers
|
| 5 |
import random
|
| 6 |
import textwrap
|
|
|
|
|
|
|
| 7 |
|
| 8 |
@st.cache
|
| 9 |
def load_model():
|
|
|
|
| 45 |
temp = st.slider('Temperature', 1, 30, 1)
|
| 46 |
num_generate = st.text_input("Enter number of sequences")
|
| 47 |
|
| 48 |
+
if st.button('Generate'):
|
| 49 |
prompt = preprocess_text(text_input, tokenizer)
|
| 50 |
result = predict_sentiment(model, prompt, temp, int(num_generate))
|
| 51 |
for i in result:
|