Update app.py

app.py

@@ -4,7 +4,8 @@ import streamlit as st
 def model(input_text):
     pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
     output=pipe("dogs1.webp")
-    return output
+    st.write("Prediction:", output[0]['label'], "| Score:", output[0]['score'])
+    #return output
 
 def main():
     st.title("Hugging Face Model Demo")
@@ -15,9 +16,9 @@ def main():
     # Create a button to trigger model inference
     if st.button("Analyze"):
         # Perform inference using the loaded model
-        result = model(input_text)
+        #result = model(input_text)
         #st.write("Prediction:", result[0]['label'], "| Score:", result[0]['score'])
-        st.write. st.write(generated_sequences[-1])
+        #st.write. st.write(generated_sequences[-1])
 
 if __name__ == "__main__":
     main()