Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import streamlit as st
|
2 |
-
from transformers import CLIPModel, pipeline, CLIPImageProcessor
|
|
|
3 |
import torch
|
4 |
from PIL import Image
|
5 |
|
@@ -52,8 +53,9 @@ if picture_file is not None:
|
|
52 |
|
53 |
#image
|
54 |
with col_l:
|
55 |
-
default_options =
|
56 |
-
options = st.text_input(label="Please enter the classes", value=default_options)
|
|
|
57 |
#options = list(options)
|
58 |
|
59 |
# button to launch compute
|
|
|
1 |
import streamlit as st
|
2 |
+
#from transformers import CLIPModel, pipeline, CLIPImageProcessor
|
3 |
+
from transformers import pipeline
|
4 |
import torch
|
5 |
from PIL import Image
|
6 |
|
|
|
53 |
|
54 |
#image
|
55 |
with col_l:
|
56 |
+
default_options = 'There is a car, There is no car'
|
57 |
+
options = st.text_input(label="Please enter the classes", value=default_options).split(',')
|
58 |
+
|
59 |
#options = list(options)
|
60 |
|
61 |
# button to launch compute
|