Commit ·
eca2bf8
1
Parent(s): 7ab9f15
removed devices
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -8,7 +8,7 @@ from transformers import BlipForConditionalGeneration, BlipProcessor
|
|
| 8 |
import requests
|
| 9 |
from PIL import Image
|
| 10 |
from transformers import Blip2Processor, Blip2ForConditionalGeneration
|
| 11 |
-
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 12 |
|
| 13 |
class EndpointHandler():
|
| 14 |
def __init__(self, path=""):
|
|
|
|
| 8 |
import requests
|
| 9 |
from PIL import Image
|
| 10 |
from transformers import Blip2Processor, Blip2ForConditionalGeneration
|
| 11 |
+
# device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 12 |
|
| 13 |
class EndpointHandler():
|
| 14 |
def __init__(self, path=""):
|