chore: Refactor CUDA device usage in app.py

app.py

@@ -106,7 +106,7 @@ def resize_without_crop(image, target_width, target_height):
 @spaces.GPU()
 def interrogator_process(x):
     image_description = wd14tagger.default_interrogator(x)
-    return image_description, image_description
+    return image_description
 
 
 @spaces.GPU()
@@ -273,7 +273,7 @@ with block:
     prompt_gen_button.click(
         fn=interrogator_process,
         inputs=[input_fg],
-        outputs=[prompt, i2v_input_text]
+        outputs=[prompt]
     ).then(lambda: [gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=False)],
            outputs=[prompt_gen_button, key_gen_button, i2v_end_btn])
 

requirements.txt

@@ -1,7 +1,6 @@
 diffusers==0.28.0
 transformers==4.41.1
 gradio==4.31.5
-bitsandbytes==0.43.1
 accelerate==0.30.1
 protobuf==3.20
 opencv-python
@@ -15,5 +14,4 @@ xformers
 onnxruntime
 av
 torchvision
-xformers
 spaces