Update app.py
Browse files
app.py
CHANGED
@@ -5,12 +5,12 @@ model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt-neo-125M")
|
|
5 |
tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
|
6 |
|
7 |
prompt = """This is a discussion between a person and a diesel locomotive.
|
8 |
-
person: Are you a diesel locomotive
|
9 |
Diesel locomotive: Yes!!
|
10 |
person: Where are you?
|
11 |
Diesel locomotive: I live in a railway station!!
|
12 |
-
person:
|
13 |
-
Diesel locomotive:
|
14 |
person: """
|
15 |
|
16 |
def my_split(s, seps):
|
|
|
5 |
tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
|
6 |
|
7 |
prompt = """This is a discussion between a person and a diesel locomotive.
|
8 |
+
person: Are you a diesel locomotive?
|
9 |
Diesel locomotive: Yes!!
|
10 |
person: Where are you?
|
11 |
Diesel locomotive: I live in a railway station!!
|
12 |
+
person: Who are you?
|
13 |
+
Diesel locomotive: I'm a machine!!
|
14 |
person: """
|
15 |
|
16 |
def my_split(s, seps):
|