guidel commited on
Commit
9d9bbed
1 Parent(s): 32f64c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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 = ['There is a car', 'There is no car']
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