jw2yang commited on
Commit
28c8be9
·
1 Parent(s): 3d0a233

override gradio version

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  import pygame
2
  import numpy as np
3
  import gradio as gr
@@ -7,11 +12,6 @@ from PIL import Image
7
  from transformers import AutoModelForCausalLM, AutoProcessor
8
  import re
9
  import random
10
- import os
11
-
12
- # add a command for installing flash-attn
13
- os.system('pip install flash-attn --no-build-isolation')
14
- os.system("pip install gradio==4.44.1")
15
 
16
  pygame.mixer.quit() # Disable sound
17
 
 
1
+ import os
2
+ # add a command for installing flash-attn
3
+ os.system('pip install flash-attn --no-build-isolation')
4
+ os.system("pip install gradio==4.44.1")
5
+
6
  import pygame
7
  import numpy as np
8
  import gradio as gr
 
12
  from transformers import AutoModelForCausalLM, AutoProcessor
13
  import re
14
  import random
 
 
 
 
 
15
 
16
  pygame.mixer.quit() # Disable sound
17