First Landing

app.py

@@ -1,34 +1,25 @@
 import os
+from http import HTTPStatus
 from fastapi.responses import StreamingResponse
 from fastapi import FastAPI, Query
-from threading import Thread
 from typing import List
+from threading import Thread
 
 import spaces
 import torch
 import uvicorn
 import time
+import numpy as np
 
-
-# Initialize Logics
-
-# 1. Install Packages  
 os.system("pip install transformers")
 os.system("pip install accelerate")
 os.system("pip install peft")
 os.system("pip install -U FlagEmbedding")
 
-
-
-# 2. Import Installed Packages
 #fmt: off
-print("⭐️")
-from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
-print("⭐️")
+from transformers import AutoModelForCausalLM, AutoTokenizer
 from FlagEmbedding import BGEM3FlagModel
-print("⭐️")
-import numpy as np
-print("⭐️")
+
 
 # 3. Initalize FastAPI App
 app = FastAPI()