Update main.py

main.py

@@ -1,44 +1,18 @@
-import gradio as gr
-from langchain_experimental.llms.ollama_functions import OllamaFunctions
+import subprocess
 
-# Initialize the Ollama model
-model = OllamaFunctions(model="gemma:7b")
-model = model.bind(
-    functions=[
-        {
-            "name": "get_current_weather",
-            "description": "Get the current weather in a given location",
-            "parameters": {
-                "type": "object",
-                "properties": {
-                    "location": {
-                        "type": "string",
-                        "description": "The city and state, e.g., San Francisco, CA",
-                    },
-                    "unit": {
-                        "type": "string",
-                        "enum": ["celsius", "fahrenheit"],
-                    },
-                },
-                "required": ["location"],
-            },
-        }
-    ],
-    function_call={"name": "get_current_weather"},
-)
+def run_command(command):
+    try:
+        result = subprocess.run(command, shell=True, check=True, text=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        print(f"Success: {result.stdout}")
+    except subprocess.CalledProcessError as e:
+        print(f"Error: {e.stderr}")
 
-def get_weather(location, unit):
-    user_input = f"{location}, {unit}"
-    result = model.invoke(user_input)
-    return result
+def main(model):
+    # Install script
+    run_command("curl -fsSL https://ollama.com/install.sh | sh")
+    
+    # Start Ollama server
+    run_command("ollama serve")
 
-iface = gr.Interface(
-    fn=get_weather,
-    inputs=[gr.Textbox(label="Location (e.g., 'San Francisco, CA')"), gr.Radio(choices=["celsius", "fahrenheit"], label="Unit")],
-    outputs=gr.Text(label="Weather Information"),
-    title="Weather Information",
-    description="Enter a location and select the unit to get the current weather.",
-    allow_flagging="never"
-)
-
-iface.launch()
+if __name__ == "__main__":
+    main()