Spaces:
Running
Running
fix
Browse files
app.py
CHANGED
@@ -4,8 +4,6 @@ import gradio as gr
|
|
4 |
from lavis.models import load_model_and_preprocess
|
5 |
from PIL import Image
|
6 |
|
7 |
-
|
8 |
-
|
9 |
def process(input_image, prompt):
|
10 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
11 |
|
|
|
4 |
from lavis.models import load_model_and_preprocess
|
5 |
from PIL import Image
|
6 |
|
|
|
|
|
7 |
def process(input_image, prompt):
|
8 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
9 |
|