Update app.py

app.py

@@ -1,7 +1,18 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+def generate_primes(text_input):
+    inputs = tokenizer(text_input, return_tensors="pt", return_attention_mask=False)
+    outputs = model.generate(**inputs, max_length=200)
+    generated_text = tokenizer.batch_decode(outputs)[0]
+    return generated_text
+
+iface = gr.Interface(
+    fn=generate_primes,
+    inputs=gr.Textbox(),
+    outputs=gr.Textbox(),
+    live=True,
+    title="Prime Generator",
+    description="Generate primes based on input text."
+)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()