Update main.py
Browse files
main.py
CHANGED
|
@@ -6,7 +6,7 @@ import os
|
|
| 6 |
app = FastAPI(title="Private LLM Proxy")
|
| 7 |
|
| 8 |
# URL of your NEW 6.7B model Space from STEP 1
|
| 9 |
-
MODEL_SPACE_URL = os.getenv("MODEL_URL", "https://
|
| 10 |
|
| 11 |
class QueryRequest(BaseModel):
|
| 12 |
prompt: str
|
|
|
|
| 6 |
app = FastAPI(title="Private LLM Proxy")
|
| 7 |
|
| 8 |
# URL of your NEW 6.7B model Space from STEP 1
|
| 9 |
+
MODEL_SPACE_URL = os.getenv("MODEL_URL", "https://truegleai-deepseek-coder-6b-api.hf.space") # <<< CHANGE THIS
|
| 10 |
|
| 11 |
class QueryRequest(BaseModel):
|
| 12 |
prompt: str
|