Spaces:

Walid-Ahmed
/

Visual_Q_and_A

Running on Zero

Walid-Ahmed commited on 4 days ago

Commit

6d34948

verified ·

1 Parent(s): c83567b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from transformers import BlipForQuestionAnswering, AutoProcessor
 from PIL import Image
 # Check device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 import torch
 from transformers import BlipForQuestionAnswering, AutoProcessor
 from PIL import Image
+import spaces
 # Check device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")