Spaces:
Sleeping
Sleeping
Rename caption.py to app.py
Browse files- caption.py → app.py +1 -1
caption.py → app.py
RENAMED
@@ -3,7 +3,7 @@ from transformers import AutoProcessor, AutoModelForCausalLM
|
|
3 |
import torch
|
4 |
from PIL import Image
|
5 |
|
6 |
-
from
|
7 |
|
8 |
with gr.Blocks() as demo:
|
9 |
image = gr.Image(type='pil', label='Image')
|
|
|
3 |
import torch
|
4 |
from PIL import Image
|
5 |
|
6 |
+
from app import predict_step
|
7 |
|
8 |
with gr.Blocks() as demo:
|
9 |
image = gr.Image(type='pil', label='Image')
|