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
|