Fix (#2)

- Fix (088953aedbc45b660577728b8b5de0f9c5614349)


Co-authored-by: hysts <[email protected]>

README.md

@@ -4,7 +4,7 @@ emoji: 🌖
 colorFrom: blue
 colorTo: pink
 sdk: gradio
-sdk_version: 3.44.4
+sdk_version: 3.50.2
 python_version: 3.10.11
 app_file: app.py
 pinned: false

app.py

@@ -18,19 +18,18 @@ if not torch.cuda.is_available():
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-MODEL_ID_OPT_6_7B = "Salesforce/blip2-opt-6.7b"
-MODEL_ID_FLAN_T5_XXL = "Salesforce/blip2-flan-t5-xxl"
-MODEL_ID = os.getenv("MODEL_ID", MODEL_ID_FLAN_T5_XXL)
+MODEL_ID = os.getenv("MODEL_ID", "Salesforce/blip2-opt-6.7b")
 
 
 processor = AutoProcessor.from_pretrained(MODEL_ID)
 
 if torch.cuda.is_available():
     model = Blip2ForConditionalGeneration.from_pretrained(MODEL_ID, device_map="auto", load_in_8bit=True)
-else: 
+else:
     model = Blip2ForConditionalGeneration.from_pretrained(MODEL_ID)
 
 
+@spaces.GPU
 def generate_caption(
     image: PIL.Image.Image,
     decoding_method: str,
@@ -54,7 +53,7 @@ def generate_caption(
     return result
 
 
-
+@spaces.GPU
 def answer_question(
     image: PIL.Image.Image,
     text: str,

requirements.txt

@@ -1,9 +1,9 @@
 accelerate==0.23.0
 bitsandbytes==0.41.1
-gradio==3.44.4
-huggingface-hub==0.17.2
-Pillow==10.0.1
+gradio==3.50.2
+huggingface-hub==0.17.3
+Pillow==10.1.0
 scipy==1.11.2
 torch==2.0.0
 torchvision==0.15.1
-transformers==4.33.2
+transformers==4.34.1