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
|