Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
a4e1286
1
Parent(s):
4444ae6
Update app.py
Browse files
app.py
CHANGED
|
@@ -223,14 +223,12 @@ def generate2(
|
|
| 223 |
|
| 224 |
return generated_list[0]
|
| 225 |
|
| 226 |
-
|
| 227 |
device = CUDA(0) if is_gpu else "cpu"
|
| 228 |
clip_model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
|
| 229 |
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
|
| 230 |
|
| 231 |
-
def inference(img,model_name):
|
| 232 |
-
is_gpu = False
|
| 233 |
-
|
| 234 |
prefix_length = 10
|
| 235 |
|
| 236 |
model = ClipCaptionModel(prefix_length)
|
|
|
|
| 223 |
|
| 224 |
return generated_list[0]
|
| 225 |
|
| 226 |
+
is_gpu = False
|
| 227 |
device = CUDA(0) if is_gpu else "cpu"
|
| 228 |
clip_model, preprocess = clip.load("ViT-B/32", device=device, jit=False)
|
| 229 |
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
|
| 230 |
|
| 231 |
+
def inference(img,model_name):
|
|
|
|
|
|
|
| 232 |
prefix_length = 10
|
| 233 |
|
| 234 |
model = ClipCaptionModel(prefix_length)
|