Upload 3 files

app.py

@@ -0,0 +1,40 @@
+import requests
+import json
+import gradio as gr
+
+url="http://localhost:11434/api/generate"
+
+headers={
+
+    'Content-Type':'application/json'
+}
+
+history=[]
+
+def generate_response(prompt):
+    history.append(prompt)
+    final_prompt="\n".join(history)
+
+    data={
+        "model":"omnicode",
+        "prompt":final_prompt,
+        "stream":False
+    }
+
+    response=requests.post(url,headers=headers,data=json.dumps(data))
+
+    if response.status_code==200:
+        response=response.text
+        data=json.loads(response)
+        actual_response=data['response']
+        return actual_response
+    else:
+        print("error:",response.text)
+
+
+interface=gr.Interface(
+    fn=generate_response,
+    inputs=gr.Textbox(lines=4,placeholder="Enter your Prompt"),
+    outputs="text"
+)
+interface.launch()

modelfile

@@ -0,0 +1,11 @@
+FROM codellama
+
+## Set the Temperature to 1 to have max creativity
+PARAMETER temperature 1
+
+## set the system prompt
+SYSTEM """
+You are a code teaching assistant named as OmniCode created 
+by Anusha K. Answer all the code related questions being asked.
+
+"""

requirements.txt

@@ -0,0 +1,7 @@
+langchain
+gradio
+
+
+
+
+