Spaces:
Paused
Paused
gmkdigitalmedia commited on
Commit ·
c52ba39
1
Parent(s): a4a97b6
Fix PyTorch Docker uid error - set USER env var before imports
Browse files- app.py +7 -0
- foundation_engine.py +7 -1
- two_llm_system_FIXED.py +7 -0
app.py
CHANGED
|
@@ -3,6 +3,13 @@ FastAPI REST API for Foundation 1.2 Clinical Trial System
|
|
| 3 |
Production-ready Docker space with proper REST endpoints
|
| 4 |
"""
|
| 5 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
from fastapi import FastAPI, HTTPException
|
| 7 |
from fastapi.middleware.cors import CORSMiddleware
|
| 8 |
from pydantic import BaseModel
|
|
|
|
| 3 |
Production-ready Docker space with proper REST endpoints
|
| 4 |
"""
|
| 5 |
|
| 6 |
+
# Fix for PyTorch/Docker uid issue - must be before any torch imports
|
| 7 |
+
import os
|
| 8 |
+
os.environ["USER"] = "user"
|
| 9 |
+
os.environ["TORCH_HOME"] = "/tmp/torch"
|
| 10 |
+
os.environ["HF_HOME"] = "/tmp/huggingface"
|
| 11 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
|
| 12 |
+
|
| 13 |
from fastapi import FastAPI, HTTPException
|
| 14 |
from fastapi.middleware.cors import CORSMiddleware
|
| 15 |
from pydantic import BaseModel
|
foundation_engine.py
CHANGED
|
@@ -3,8 +3,14 @@ Foundation 1.2
|
|
| 3 |
Clinical trial query system with 355M foundation model
|
| 4 |
"""
|
| 5 |
|
| 6 |
-
|
| 7 |
import os
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
from pathlib import Path
|
| 9 |
import pickle
|
| 10 |
import numpy as np
|
|
|
|
| 3 |
Clinical trial query system with 355M foundation model
|
| 4 |
"""
|
| 5 |
|
| 6 |
+
# Fix for PyTorch/Docker uid issue - must be before any torch imports
|
| 7 |
import os
|
| 8 |
+
os.environ["USER"] = "user"
|
| 9 |
+
os.environ["TORCH_HOME"] = "/tmp/torch"
|
| 10 |
+
os.environ["HF_HOME"] = "/tmp/huggingface"
|
| 11 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
|
| 12 |
+
|
| 13 |
+
import gradio as gr
|
| 14 |
from pathlib import Path
|
| 15 |
import pickle
|
| 16 |
import numpy as np
|
two_llm_system_FIXED.py
CHANGED
|
@@ -4,6 +4,13 @@ Works with existing data structure: List[Tuple[int, str]]
|
|
| 4 |
Keeps BM25 + semantic hybrid search intact
|
| 5 |
"""
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
import torch
|
| 8 |
from transformers import GPT2LMHeadModel, GPT2TokenizerFast, AutoTokenizer, AutoModelForCausalLM
|
| 9 |
import logging
|
|
|
|
| 4 |
Keeps BM25 + semantic hybrid search intact
|
| 5 |
"""
|
| 6 |
|
| 7 |
+
# Fix for PyTorch/Docker uid issue - must be before any torch imports
|
| 8 |
+
import os
|
| 9 |
+
os.environ.setdefault("USER", "user")
|
| 10 |
+
os.environ.setdefault("TORCH_HOME", "/tmp/torch")
|
| 11 |
+
os.environ.setdefault("HF_HOME", "/tmp/huggingface")
|
| 12 |
+
os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/transformers")
|
| 13 |
+
|
| 14 |
import torch
|
| 15 |
from transformers import GPT2LMHeadModel, GPT2TokenizerFast, AutoTokenizer, AutoModelForCausalLM
|
| 16 |
import logging
|