jbilcke-hf HF staff commited on
Commit
da1838e
β€’
1 Parent(s): df46f1b
Files changed (2) hide show
  1. src/getGradioApp.mts +1 -0
  2. src/getStreamlitApp.mts +3 -1
src/getGradioApp.mts CHANGED
@@ -22,6 +22,7 @@ You MUST use the following Python modules:
22
  You are free to use (if necessary) the following Python modules:
23
  - numpy
24
  - matplotlib
 
25
  - diffusers
26
  - transformers
27
 
 
22
  You are free to use (if necessary) the following Python modules:
23
  - numpy
24
  - matplotlib
25
+ - pytorch
26
  - diffusers
27
  - transformers
28
 
src/getStreamlitApp.mts CHANGED
@@ -18,8 +18,10 @@ export function getStreamlitApp(prompt: string) {
18
 
19
  Please limit yourself to the following Python modules:
20
  - numpy
21
- - streamlit
22
  - matplotlib
 
 
 
23
 
24
  Don't forget to write a README.md with the following header:
25
  \`\`\`
 
18
 
19
  Please limit yourself to the following Python modules:
20
  - numpy
 
21
  - matplotlib
22
+ - pytorch
23
+ - diffusers
24
+ - transformers
25
 
26
  Don't forget to write a README.md with the following header:
27
  \`\`\`