Spaces:
Runtime error
Runtime error
Logan Zoellner
commited on
Commit
·
aa35052
1
Parent(s):
78317e6
cleanup slightly
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ import requests
|
|
4 |
import os
|
5 |
import re
|
6 |
import random
|
|
|
7 |
|
8 |
# GPT-J-6B API
|
9 |
API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
|
@@ -21,9 +22,6 @@ examples = [["river"], ["night"], ["trees"],["table"],["laughs"]]
|
|
21 |
|
22 |
def npc_randomize():
|
23 |
#name is a random combination of syllables
|
24 |
-
vowels = list("aeiou")
|
25 |
-
constants = list("bcdfghjklmnpqrstvwxyz")
|
26 |
-
seperators=list("-'")
|
27 |
name =""
|
28 |
for i in range(random.randint(2,4)):
|
29 |
name += random.choice(constants)
|
@@ -32,109 +30,15 @@ def npc_randomize():
|
|
32 |
name += random.choice(constants)
|
33 |
if random.random()<0.1:
|
34 |
name += random.choice(seperators)
|
35 |
-
|
36 |
#capitalize first letter
|
37 |
name = name[0].upper() + name[1:]
|
38 |
-
|
39 |
-
races="""Dwarf
|
40 |
-
Elf
|
41 |
-
Halfling
|
42 |
-
Human
|
43 |
-
Dragonborn
|
44 |
-
Gnome
|
45 |
-
Half-elf
|
46 |
-
Half-orc
|
47 |
-
Tiefling
|
48 |
-
Aarakocra
|
49 |
-
Genasi
|
50 |
-
Goliath""".split("\n")
|
51 |
-
|
52 |
-
races=[x.strip() for x in races]
|
53 |
-
|
54 |
race=random.choice(races)
|
55 |
-
|
56 |
-
print("foo",races,race)
|
57 |
-
|
58 |
-
|
59 |
-
classes="""Barbarian
|
60 |
-
Bard
|
61 |
-
Cleric
|
62 |
-
Druid
|
63 |
-
Fighter
|
64 |
-
Monk
|
65 |
-
Paladin
|
66 |
-
Ranger
|
67 |
-
Rogue
|
68 |
-
Sorcerer
|
69 |
-
Warlock
|
70 |
-
Wizard""".split("\n")
|
71 |
-
|
72 |
-
classes=[x.strip() for x in classes]
|
73 |
-
|
74 |
characterClass=random.choice(classes)
|
75 |
-
|
76 |
pronoun=random.choices(["he","she","they"],weights=[0.45,0.45,0.1],k=1)[0]
|
77 |
-
|
78 |
return name,race,characterClass,pronoun
|
79 |
|
80 |
|
81 |
-
|
82 |
-
|
83 |
def genericDescription():
|
84 |
-
colors="""red
|
85 |
-
blue
|
86 |
-
green
|
87 |
-
yellow
|
88 |
-
orange
|
89 |
-
purple
|
90 |
-
pink
|
91 |
-
brown
|
92 |
-
black
|
93 |
-
white""".split("\n")
|
94 |
-
colors=[x.strip() for x in colors]
|
95 |
-
outfits="""shirt
|
96 |
-
pair of pants
|
97 |
-
pair of shoes
|
98 |
-
hat
|
99 |
-
pair of glasses
|
100 |
-
backpack
|
101 |
-
belt
|
102 |
-
tie
|
103 |
-
cloak
|
104 |
-
robe
|
105 |
-
chain mail vest
|
106 |
-
suit of plate armor
|
107 |
-
suit of leather armor
|
108 |
-
suit of studded leather armor
|
109 |
-
suit of scale armor
|
110 |
-
suit of chain mail armor
|
111 |
-
suit of ring mail armor
|
112 |
-
""".split("\n")
|
113 |
-
outfits=[x.strip() for x in outfits]
|
114 |
-
weapons="""sword
|
115 |
-
dagger
|
116 |
-
mace
|
117 |
-
axe
|
118 |
-
polearm
|
119 |
-
bow
|
120 |
-
crossbow
|
121 |
-
sling
|
122 |
-
club
|
123 |
-
flail
|
124 |
-
warhammer
|
125 |
-
morningstar
|
126 |
-
halberd
|
127 |
-
war pick
|
128 |
-
war sickle
|
129 |
-
war hammer""".split("\n")
|
130 |
-
weapons=[x.strip() for x in weapons]
|
131 |
-
|
132 |
-
objects="""shield
|
133 |
-
lantern
|
134 |
-
sack
|
135 |
-
severed head
|
136 |
-
crystal""".split("\n")
|
137 |
-
objects=[x.strip() for x in objects]
|
138 |
|
139 |
desc=" wears a {color} {outfit}".format(color=random.choice(colors),outfit=random.choice(outfits))
|
140 |
if random.random()<0.5:
|
|
|
4 |
import os
|
5 |
import re
|
6 |
import random
|
7 |
+
from words import *
|
8 |
|
9 |
# GPT-J-6B API
|
10 |
API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
|
|
|
22 |
|
23 |
def npc_randomize():
|
24 |
#name is a random combination of syllables
|
|
|
|
|
|
|
25 |
name =""
|
26 |
for i in range(random.randint(2,4)):
|
27 |
name += random.choice(constants)
|
|
|
30 |
name += random.choice(constants)
|
31 |
if random.random()<0.1:
|
32 |
name += random.choice(seperators)
|
|
|
33 |
#capitalize first letter
|
34 |
name = name[0].upper() + name[1:]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
race=random.choice(races)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
characterClass=random.choice(classes)
|
|
|
37 |
pronoun=random.choices(["he","she","they"],weights=[0.45,0.45,0.1],k=1)[0]
|
|
|
38 |
return name,race,characterClass,pronoun
|
39 |
|
40 |
|
|
|
|
|
41 |
def genericDescription():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
42 |
|
43 |
desc=" wears a {color} {outfit}".format(color=random.choice(colors),outfit=random.choice(outfits))
|
44 |
if random.random()<0.5:
|
words.py
ADDED
@@ -0,0 +1,93 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
vowels = list("aeiou")
|
3 |
+
constants = list("bcdfghjklmnpqrstvwxyz")
|
4 |
+
seperators=list("-'")
|
5 |
+
|
6 |
+
races="""Dwarf
|
7 |
+
Elf
|
8 |
+
Halfling
|
9 |
+
Human
|
10 |
+
Dragonborn
|
11 |
+
Gnome
|
12 |
+
Half-elf
|
13 |
+
Half-orc
|
14 |
+
Tiefling
|
15 |
+
Aarakocra
|
16 |
+
Genasi
|
17 |
+
Goliath""".split("\n")
|
18 |
+
|
19 |
+
races=[x.strip() for x in races]
|
20 |
+
|
21 |
+
|
22 |
+
|
23 |
+
|
24 |
+
classes="""Barbarian
|
25 |
+
Bard
|
26 |
+
Cleric
|
27 |
+
Druid
|
28 |
+
Fighter
|
29 |
+
Monk
|
30 |
+
Paladin
|
31 |
+
Ranger
|
32 |
+
Rogue
|
33 |
+
Sorcerer
|
34 |
+
Warlock
|
35 |
+
Wizard""".split("\n")
|
36 |
+
classes=[x.strip() for x in classes]
|
37 |
+
|
38 |
+
colors="""red
|
39 |
+
blue
|
40 |
+
green
|
41 |
+
yellow
|
42 |
+
orange
|
43 |
+
purple
|
44 |
+
pink
|
45 |
+
brown
|
46 |
+
black
|
47 |
+
white""".split("\n")
|
48 |
+
colors=[x.strip() for x in colors]
|
49 |
+
|
50 |
+
outfits="""shirt
|
51 |
+
pair of pants
|
52 |
+
pair of shoes
|
53 |
+
hat
|
54 |
+
pair of glasses
|
55 |
+
backpack
|
56 |
+
belt
|
57 |
+
tie
|
58 |
+
cloak
|
59 |
+
robe
|
60 |
+
chain mail vest
|
61 |
+
suit of plate armor
|
62 |
+
suit of leather armor
|
63 |
+
suit of studded leather armor
|
64 |
+
suit of scale armor
|
65 |
+
suit of chain mail armor
|
66 |
+
suit of ring mail armor
|
67 |
+
""".split("\n")
|
68 |
+
outfits=[x.strip() for x in outfits]
|
69 |
+
|
70 |
+
weapons="""sword
|
71 |
+
dagger
|
72 |
+
mace
|
73 |
+
axe
|
74 |
+
polearm
|
75 |
+
bow
|
76 |
+
crossbow
|
77 |
+
sling
|
78 |
+
club
|
79 |
+
flail
|
80 |
+
warhammer
|
81 |
+
morningstar
|
82 |
+
halberd
|
83 |
+
war pick
|
84 |
+
war sickle
|
85 |
+
war hammer""".split("\n")
|
86 |
+
weapons=[x.strip() for x in weapons]
|
87 |
+
|
88 |
+
objects="""shield
|
89 |
+
lantern
|
90 |
+
sack
|
91 |
+
severed head
|
92 |
+
crystal""".split("\n")
|
93 |
+
objects=[x.strip() for x in objects]
|