[Admin maintenance] Support new ZeroGPU hardware (#6)

- [Admin maintenance] Support new ZeroGPU hardware (c949f9d8647ea9f8a13a1f5e05114925e0efb329)

README.md

@@ -8,6 +8,7 @@ sdk_version: 5.38.2
 app_file: app.py
 pinned: false
 license: apache-2.0
+python_version: "3.10"
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,12 +1,10 @@
+import spaces
 import gradio as gr
 from transformers import AutoProcessor, AutoModelForImageTextToText, TextIteratorStreamer
 from threading import Thread
 import re
 import time
 import torch
-import spaces
-import subprocess
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 from io import BytesIO
 

requirements.txt

@@ -1,10 +1,11 @@
-torch
-torchvision
-pyav
+torch==2.8.0
+torchvision==0.23.0
+av
 accelerate
 huggingface_hub
 gradio
 transformers
 spaces
 num2words
-decord
+decord
+https://huggingface.co/datasets/multimodalart/zerogpu-blackwell-wheels/resolve/main/wheels/pt28-cu128-cp310/flash_attn-2.8.3-cp310-cp310-linux_x86_64.whl