Spaces:
Running
Running
Formatted with isort.
Browse filesrefactor: reorganize imports across multiple files for consistency
- app.py +1 -0
- optipfair_backend.py +1 -0
- optipfair_frontend.py +3 -2
- routers/visualize.py +7 -11
- schemas/visualize.py +2 -1
- utils/visualize_pca.py +5 -5
app.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
import threading
|
| 3 |
import time
|
|
|
|
| 4 |
import uvicorn
|
| 5 |
from optipfair_backend import app as fastapi_app
|
| 6 |
from optipfair_frontend import create_interface
|
|
|
|
| 1 |
import os
|
| 2 |
import threading
|
| 3 |
import time
|
| 4 |
+
|
| 5 |
import uvicorn
|
| 6 |
from optipfair_backend import app as fastapi_app
|
| 7 |
from optipfair_frontend import create_interface
|
optipfair_backend.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
from fastapi import FastAPI
|
| 2 |
from fastapi.middleware.cors import CORSMiddleware # ← NUEVO: Para CORS
|
|
|
|
| 3 |
from routers.visualize import router as visualize_router
|
| 4 |
|
| 5 |
# Create FastAPI app with HF Spaces configuration
|
|
|
|
| 1 |
from fastapi import FastAPI
|
| 2 |
from fastapi.middleware.cors import CORSMiddleware # ← NUEVO: Para CORS
|
| 3 |
+
|
| 4 |
from routers.visualize import router as visualize_router
|
| 5 |
|
| 6 |
# Create FastAPI app with HF Spaces configuration
|
optipfair_frontend.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
-
import gradio as gr
|
| 2 |
-
import requests
|
| 3 |
import logging
|
| 4 |
import os
|
| 5 |
import tempfile
|
| 6 |
|
|
|
|
|
|
|
|
|
|
| 7 |
# Configure logging
|
| 8 |
logging.basicConfig(level=logging.INFO)
|
| 9 |
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
|
| 1 |
import logging
|
| 2 |
import os
|
| 3 |
import tempfile
|
| 4 |
|
| 5 |
+
import gradio as gr
|
| 6 |
+
import requests
|
| 7 |
+
|
| 8 |
# Configure logging
|
| 9 |
logging.basicConfig(level=logging.INFO)
|
| 10 |
logger = logging.getLogger(__name__)
|
routers/visualize.py
CHANGED
|
@@ -1,18 +1,14 @@
|
|
| 1 |
# routers/visualize.py
|
| 2 |
-
import os
|
| 3 |
import logging
|
|
|
|
|
|
|
| 4 |
from fastapi import APIRouter, HTTPException
|
| 5 |
from fastapi.responses import FileResponse
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
)
|
| 11 |
-
from utils.visualize_pca import (
|
| 12 |
-
run_visualize_pca,
|
| 13 |
-
run_visualize_mean_diff,
|
| 14 |
-
run_visualize_heatmap,
|
| 15 |
-
)
|
| 16 |
|
| 17 |
logger = logging.getLogger(__name__)
|
| 18 |
logger.setLevel(logging.INFO)
|
|
|
|
| 1 |
# routers/visualize.py
|
|
|
|
| 2 |
import logging
|
| 3 |
+
import os
|
| 4 |
+
|
| 5 |
from fastapi import APIRouter, HTTPException
|
| 6 |
from fastapi.responses import FileResponse
|
| 7 |
+
|
| 8 |
+
from schemas.visualize import (VisualizeHeatmapRequest,
|
| 9 |
+
VisualizeMeanDiffRequest, VisualizePCARequest)
|
| 10 |
+
from utils.visualize_pca import (run_visualize_heatmap,
|
| 11 |
+
run_visualize_mean_diff, run_visualize_pca)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
logger = logging.getLogger(__name__)
|
| 14 |
logger.setLevel(logging.INFO)
|
schemas/visualize.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
# schemas/visualize.py
|
|
|
|
|
|
|
| 2 |
from pydantic import BaseModel, field_validator
|
| 3 |
-
from typing import List, Optional, Union, Tuple
|
| 4 |
|
| 5 |
|
| 6 |
class VisualizePCARequest(BaseModel):
|
|
|
|
| 1 |
# schemas/visualize.py
|
| 2 |
+
from typing import List, Optional, Tuple, Union
|
| 3 |
+
|
| 4 |
from pydantic import BaseModel, field_validator
|
|
|
|
| 5 |
|
| 6 |
|
| 7 |
class VisualizePCARequest(BaseModel):
|
utils/visualize_pca.py
CHANGED
|
@@ -1,16 +1,16 @@
|
|
| 1 |
# utils/visualize_pca.py
|
|
|
|
| 2 |
import os
|
| 3 |
import tempfile
|
| 4 |
-
import logging
|
| 5 |
from functools import lru_cache
|
| 6 |
-
from typing import
|
| 7 |
|
|
|
|
| 8 |
import torch
|
| 9 |
-
from optipfair.bias import
|
|
|
|
| 10 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 11 |
|
| 12 |
-
import matplotlib
|
| 13 |
-
|
| 14 |
matplotlib.use("Agg") # Use 'Agg' backend for non-GUI environments
|
| 15 |
|
| 16 |
logger = logging.getLogger(__name__)
|
|
|
|
| 1 |
# utils/visualize_pca.py
|
| 2 |
+
import logging
|
| 3 |
import os
|
| 4 |
import tempfile
|
|
|
|
| 5 |
from functools import lru_cache
|
| 6 |
+
from typing import List, Optional, Tuple, Union
|
| 7 |
|
| 8 |
+
import matplotlib
|
| 9 |
import torch
|
| 10 |
+
from optipfair.bias import (visualize_heatmap, visualize_mean_differences,
|
| 11 |
+
visualize_pca)
|
| 12 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 13 |
|
|
|
|
|
|
|
| 14 |
matplotlib.use("Agg") # Use 'Agg' backend for non-GUI environments
|
| 15 |
|
| 16 |
logger = logging.getLogger(__name__)
|