Update main.py

main.py

@@ -1,18 +1,17 @@
-import subprocess
+import requests
+import json
 
-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}")
+reqUrl = "http://localhost:11434/api/pull"
 
-def main():
-    # Install script
-    run_command("curl -fsSL https://ollama.com/install.sh | sh")
-    
-    # Start Ollama server
-    run_command("ollama serve")
+headersList = {
+ "Accept": "*/*",
+ "Content-Type": "application/json" 
+}
 
-if __name__ == "__main__":
-    main()
+payload = json.dumps({
+  "name": "gemma:7b"
+})
+
+response = requests.request("POST", reqUrl, data=payload,  headers=headersList)
+
+print(response.text)