Update app.py

app.py

@@ -1,26 +1,24 @@
 import gradio as gr
 from transformers import pipeline
 
-
+# Load your text generation model from Hugging Face
 model_name = "curiouscurrent/omnicode"
 text_generator = pipeline("text-generation", model=model_name)
 
-history = []
-
-def generate_response(prompt):
-    history.append(prompt)
-    final_prompt = "\n".join(history)
-    
+def generate_response(input_prompt):
     # Generate response
-    response = text_generator(final_prompt, max_length=100)[0]['generated_text']
-    
+    response = text_generator(input_prompt, max_length=100)[0]['generated_text']
     return response
 
-interface = gr.Interface(
-    fn=generate_response,
-    inputs=gr.inputs.Textarea(lines=4, placeholder="Enter your Prompt"),
-    outputs="text",
-    title="Text Generation App",
-    description="Generate text based on the input prompt."
-)
-interface.launch()
+# Create Gradio interface
+input_prompt = gr.inputs.Textbox(lines=5, label="Input Prompt")
+output_text = gr.outputs.Textbox(label="Response")
+
+gr.Interface(
+    generate_response,
+    inputs=input_prompt,
+    outputs=output_text,
+    title="OmniCode",
+    description="Multi Programming coding assistant.",
+    theme="compact"
+).launch()