Update app.py

app.py

@@ -1,3 +1,6 @@
+import gradio as gr
+
+# Importing the required libraries
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load model directly
@@ -24,11 +27,11 @@ def generate_response(prompt, max_length=150, temperature=1.0):
     return response
 
 
-if __name__ == "__main__":
-    while True:
-        user_input = input("You: ")
-        if not user_input:  # Check if user input is empty
-            print("Exiting OmniCode. Thank you for using me!")
-            break
-        response = generate_response(user_input)
-        print("OmniCode:", response)
+# Create Gradio interface
+def chat_with_omnicode(prompt):
+    response = generate_response(prompt)
+    return response
+
+
+iface = gr.Interface(fn=chat_with_omnicode, inputs="text", outputs="text", title="OmniCode")
+iface.launch()