Spaces:
Runtime error
Runtime error
Johannes
commited on
Commit
•
ed1dd9b
1
Parent(s):
ea23bc4
add accelerate and update types
Browse files- app.py +3 -2
- requirements.txt +2 -1
app.py
CHANGED
@@ -8,6 +8,7 @@ import os
|
|
8 |
from xml.etree import ElementTree
|
9 |
import random
|
10 |
import re
|
|
|
11 |
|
12 |
|
13 |
pokemon_types = ["Normal",
|
@@ -168,7 +169,7 @@ with block:
|
|
168 |
else:
|
169 |
return pok_type
|
170 |
|
171 |
-
def update_prompt_link(new_link: str, pok_type: str, prompt_ideas:
|
172 |
global paper_name
|
173 |
paper_name = get_paper_name(new_link)
|
174 |
pok_type = resolve_poke_type(pok_type)
|
@@ -177,7 +178,7 @@ with block:
|
|
177 |
|
178 |
return build_prompt_text(paper_name, pok_type, prompt_ideas)
|
179 |
|
180 |
-
def update_prompt_type(paper_link: str, pok_type: str, prompt_ideas:
|
181 |
global paper_name
|
182 |
if paper_name is None:
|
183 |
paper_name = get_paper_name(paper_link)
|
|
|
8 |
from xml.etree import ElementTree
|
9 |
import random
|
10 |
import re
|
11 |
+
from typing import List
|
12 |
|
13 |
|
14 |
pokemon_types = ["Normal",
|
|
|
169 |
else:
|
170 |
return pok_type
|
171 |
|
172 |
+
def update_prompt_link(new_link: str, pok_type: str, prompt_ideas: List[str]):
|
173 |
global paper_name
|
174 |
paper_name = get_paper_name(new_link)
|
175 |
pok_type = resolve_poke_type(pok_type)
|
|
|
178 |
|
179 |
return build_prompt_text(paper_name, pok_type, prompt_ideas)
|
180 |
|
181 |
+
def update_prompt_type(paper_link: str, pok_type: str, prompt_ideas: List[str]):
|
182 |
global paper_name
|
183 |
if paper_name is None:
|
184 |
paper_name = get_paper_name(paper_link)
|
requirements.txt
CHANGED
@@ -4,4 +4,5 @@ diffusers
|
|
4 |
transformers
|
5 |
scipy
|
6 |
ftfy
|
7 |
-
datasets
|
|
|
|
4 |
transformers
|
5 |
scipy
|
6 |
ftfy
|
7 |
+
datasets
|
8 |
+
accelerate
|