Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import torch.nn as nn
|
|
7 |
from transformers.activations import get_activation
|
8 |
from transformers import AutoTokenizer, AutoModelWithLMHead, AutoModelForCausalLM
|
9 |
|
10 |
-
st.title('GPT2:')
|
11 |
#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
12 |
number_of_outputs = st.sidebar.slider("Number of Outputs", 50, 350)
|
13 |
|
|
|
7 |
from transformers.activations import get_activation
|
8 |
from transformers import AutoTokenizer, AutoModelWithLMHead, AutoModelForCausalLM
|
9 |
|
10 |
+
st.title('GPT2: To see all prompt outlines: https://huggingface.co/BigSalmon/InformalToFormalLincoln40')
|
11 |
#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
12 |
number_of_outputs = st.sidebar.slider("Number of Outputs", 50, 350)
|
13 |
|