Update app.py

app.py

@@ -1,28 +1,11 @@
 import gradio as gr
-import torch
-from transformers import AutoModel, AutoTokenizer
+from transformers import pipeline
 
-# Load model and tokenizer
 model_name = "Genius-Society/insecta"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModel.from_pretrained(model_name)
+detection_pipeline = pipeline("object-detection", model=model_name)
 
-# Define inference function
-def generate_response(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt")  # Tokenize input text
-    with torch.no_grad():  # Disable gradients for inference
-        output = model(**inputs)  
+def detect_objects(image):
+    return detection_pipeline(image)
 
-    return str(output)  # Convert output to string for display
-
-# Create Gradio interface
-iface = gr.Interface(
-    fn=generate_response,
-    inputs="text",
-    outputs="text",
-    title="Insecta AI Model",
-    description="Generate responses using Genius-Society/Insecta hosted on Hugging Face Spaces."
-)
-
-if __name__ == "__main__":
-    iface.launch()
+iface = gr.Interface(fn=detect_objects, inputs="image", outputs="json")
+iface.launch()