Update app.py

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
+from transformers import pipeline
 
-def predict(image):
-    # Replace with your model inference code
-    return "Insect detected: Example species"
+# Load model
+model_name = "Genius-Society/insecta"
+insecta_pipeline = pipeline("text-generation", model=model_name)
 
-iface = gr.Interface(fn=predict, inputs=gr.Image(type="pil"), outputs="text")
+def generate_response(prompt):
+    return insecta_pipeline(prompt, max_length=100)[0]['generated_text']
 
+# Gradio interface
+iface = gr.Interface(fn=generate_response, inputs="text", outputs="text")
 iface.launch()