Johannes commited on
Commit
ed1dd9b
1 Parent(s): ea23bc4

add accelerate and update types

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  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: list[str]):
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: list[str]):
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