Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
fd79eb2
1
Parent(s):
3e5a168
Playing with models.
Browse files
app.py
CHANGED
@@ -4,6 +4,19 @@ import torch
|
|
4 |
from transformers import AutoTokenizer
|
5 |
from transformers import pipeline
|
6 |
from optimum.onnxruntime import ORTModelForSequenceClassification
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
|
8 |
# "xenova/mobilebert-uncased-mnli" "typeform/mobilebert-uncased-mnli" Fast but small--same as bundled in Statosphere
|
9 |
# "xenova/deberta-v3-base-tasksource-nli" Not impressed
|
|
|
4 |
from transformers import AutoTokenizer
|
5 |
from transformers import pipeline
|
6 |
from optimum.onnxruntime import ORTModelForSequenceClassification
|
7 |
+
from fastapi import FastAPI
|
8 |
+
from fastapi.middleware.cors import CORSMiddleware
|
9 |
+
|
10 |
+
# CORS Config
|
11 |
+
app = FastAPI()
|
12 |
+
|
13 |
+
app.add_middleware(
|
14 |
+
CORSMiddleware,
|
15 |
+
allow_origins=["https://jhuhman.com"], #["https://statosphere-3704059fdd7e.c5v4v4jx6pq5.win"],
|
16 |
+
allow_credentials=True,
|
17 |
+
allow_methods=["*"],
|
18 |
+
allow_headers=["*"],
|
19 |
+
)
|
20 |
|
21 |
# "xenova/mobilebert-uncased-mnli" "typeform/mobilebert-uncased-mnli" Fast but small--same as bundled in Statosphere
|
22 |
# "xenova/deberta-v3-base-tasksource-nli" Not impressed
|