Update app.py

app.py

@@ -1,3 +1,26 @@
+
+from transformers import pipeline
+
+model(input_text):
+    pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
+    output=pipe("dogs1.webp")
+    return output
+
+def main():
+    st.title("Hugging Face Model Demo")
+
+    # Create an input text box
+    input_text = st.text_input("Enter your text", "")
+
+    # Create a button to trigger model inference
+    if st.button("Analyze"):
+        # Perform inference using the loaded model
+        result = model(input_text)
+        st.write("Prediction:", result[0]['label'], "| Score:", result[0]['score'])
+
+if __name__ == "__main__":
+    main()
+'''
 import streamlit as st
 
 st.success("welcome")  #green box
@@ -44,3 +67,4 @@ if ok == True:           #user clicked
                 f"Detected {model.config.id2label[label.item()]} with confidence "
                 f"{round(score.item(), 3)} at location {box}"
         )
+'''