subh71 commited on
Commit
0d8e676
·
verified ·
1 Parent(s): bd9e6c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
 
2
  from transformers import pipeline
3
  import streamlit as st
4
- def model(input_text):
5
  pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
6
  output=pipe("dogs1.webp")
7
  st.write("Prediction:", output[0]['label'], "| Score:", output[0]['score'])
@@ -16,7 +16,7 @@ def main():
16
  # Create a button to trigger model inference
17
  if st.button("Analyze"):
18
  # Perform inference using the loaded model
19
- model(input_text)
20
  #st.write("Prediction:", result[0]['label'], "| Score:", result[0]['score'])
21
  #st.write. st.write(generated_sequences[-1])
22
 
 
1
 
2
  from transformers import pipeline
3
  import streamlit as st
4
+ def model():
5
  pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
6
  output=pipe("dogs1.webp")
7
  st.write("Prediction:", output[0]['label'], "| Score:", output[0]['score'])
 
16
  # Create a button to trigger model inference
17
  if st.button("Analyze"):
18
  # Perform inference using the loaded model
19
+ model()
20
  #st.write("Prediction:", result[0]['label'], "| Score:", result[0]['score'])
21
  #st.write. st.write(generated_sequences[-1])
22