Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 from typing import List
 import requests
 import json
+import subprocess
+from multiprocessing import Process
 
 def get_text_embedding(text: str, model: str = "mxbai-embed-large", api_url: str = "http://localhost:11434/api/embeddings") -> List[float]:
     """
@@ -39,6 +41,19 @@ def process_text_to_embedding(text: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
+def run_ollama_serve():
+    subprocess.run(["ollama", "serve"], check=True)
+
+# Create processes
+serve_process = Process(target=run_ollama_serve)
+
+# Start processes
+serve_process.start()
+
+subprocess.run(["sudo", "apt", "install", "-y", "pciutils", "lshw"], check=True)
+subprocess.run(["curl", "-fsSL", "https://ollama.com/install.sh", "|", "sh"], shell=True, check=True)
+subprocess.run(["ollama", "pull", "snowflake-arctic-embed2"], check=True)
+
 # Define the Gradio interface
 def main():
     title = "Text Embedding Generator"