Update app.py

app.py

@@ -1,41 +1,26 @@
-import requests
-import json
 import gradio as gr
+from transformers import pipeline
 
 
-
-url = "https://huggingface.co/curiouscurrent/omnicode"
-
-headers = {
-    'Content-Type': 'application/json'
-}
+model_name = "curiouscurrent/omnicode"
+text_generator = pipeline("text-generation", model=model_name)
 
 history = []
 
 def generate_response(prompt):
-    global history
     history.append(prompt)
     final_prompt = "\n".join(history)
-    data = {
-        "model": "curiouscurrent/omnicode",
-        "prompt": final_prompt,
-        "stream": False
-    }
-    response = requests.post(url, headers=headers, data=json.dumps(data))
-
-    if response.status_code == 200:
-        response = response.json()
-        actual_response = response['response']
-        return actual_response
-    else:
-        print("error:", response.text)
-
+    
+    # Generate response
+    response = text_generator(final_prompt, max_length=100)[0]['generated_text']
+    
+    return response
 
 interface = gr.Interface(
     fn=generate_response,
-    inputs=gr.Textbox(lines=4, placeholder="Enter your Prompt"),
+    inputs=gr.inputs.Textarea(lines=4, placeholder="Enter your Prompt"),
     outputs="text",
-    title="Omnicode",
-    description="A code generation model trained on a variety of programming languages."
+    title="Text Generation App",
+    description="Generate text based on the input prompt."
 )
 interface.launch()