Update main.py

main.py

@@ -1,5 +1,3 @@
-import requests
-import json
 import subprocess
 
 def run_command(command):
@@ -9,19 +7,12 @@ def run_command(command):
     except subprocess.CalledProcessError as e:
         print(f"Error: {e.stderr}")
 
-run_command("ollama serve")
+def main():
+    # Install script
+    run_command("curl -fsSL https://ollama.com/install.sh | sh")
+    
+    # Start Ollama server
+    run_command("ollama serve")
 
-reqUrl = "http://localhost:11434/api/pull"
-
-headersList = {
- "Accept": "*/*",
- "Content-Type": "application/json" 
-}
-
-payload = json.dumps({
-  "name": "gemma:7b"
-})
-
-response = requests.request("POST", reqUrl, data=payload,  headers=headersList)
-
-print(response.text)
+if __name__ == "__main__":
+    main()