update

app.py

@@ -24,7 +24,8 @@ import subprocess
 import importlib
 import sys
 
-@spaces.GPU
+
+
 os.system('nvidia-smi')
 os.system('apt update -y && apt-get install -y apt-utils && apt install -y unzip')
 os.system('pip install flash-attn --no-build-isolation')
@@ -191,4 +192,5 @@ def main():
 
 if __name__ == '__main__':
 	# subprocess.run(['bash', 'install.sh'], shell=True)
+	@spaces.GPU
 	main()