Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
46aa75d
1
Parent(s):
9231215
Experimenting with few-shot classification.
Browse files
app.py
CHANGED
|
@@ -8,6 +8,7 @@ from optimum.onnxruntime import ORTModelForFeatureExtraction
|
|
| 8 |
from fastapi import FastAPI
|
| 9 |
from fastapi.middleware.cors import CORSMiddleware
|
| 10 |
from setfit import SetFitModel
|
|
|
|
| 11 |
|
| 12 |
# CORS Config
|
| 13 |
app = FastAPI()
|
|
|
|
| 8 |
from fastapi import FastAPI
|
| 9 |
from fastapi.middleware.cors import CORSMiddleware
|
| 10 |
from setfit import SetFitModel
|
| 11 |
+
from setfit.exporters.utils import mean_pooling
|
| 12 |
|
| 13 |
# CORS Config
|
| 14 |
app = FastAPI()
|