Lord-Raven commited on
Commit
fd79eb2
·
1 Parent(s): 3e5a168

Playing with models.

Browse files
Files changed (1) hide show
  1. app.py +13 -0
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