cody82 commited on
Commit
8ed0339
·
verified ·
1 Parent(s): 48ffe3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -6,7 +6,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
6
  from fastapi import FastAPI
7
  from fastapi.middleware.cors import CORSMiddleware
8
  from pydantic import BaseModel
9
- import uvicorn
10
 
11
  model_id = "sberbank-ai/rugpt3medium_based_on_gpt2"
12
 
@@ -45,13 +44,11 @@ def respond(message: str) -> str:
45
 
46
  return answer
47
 
48
- # FastAPI app
49
  app = FastAPI(title="Иннополис бот API")
50
 
51
- # Чтобы Unity или браузеры могли обращаться, разрешим CORS (подстрой по своему домену)
52
  app.add_middleware(
53
  CORSMiddleware,
54
- allow_origins=["*"], # или укажи нужный адрес, например ["http://localhost:3000"]
55
  allow_credentials=True,
56
  allow_methods=["*"],
57
  allow_headers=["*"],
@@ -69,4 +66,5 @@ def ask_question(request: QuestionRequest):
69
  return {"answer": answer}
70
 
71
  if __name__ == "__main__":
72
- iface.launch(server_name="0.0.0.0", server_port=8000)
 
 
6
  from fastapi import FastAPI
7
  from fastapi.middleware.cors import CORSMiddleware
8
  from pydantic import BaseModel
 
9
 
10
  model_id = "sberbank-ai/rugpt3medium_based_on_gpt2"
11
 
 
44
 
45
  return answer
46
 
 
47
  app = FastAPI(title="Иннополис бот API")
48
 
 
49
  app.add_middleware(
50
  CORSMiddleware,
51
+ allow_origins=["*"], # можно указать конкретный домен
52
  allow_credentials=True,
53
  allow_methods=["*"],
54
  allow_headers=["*"],
 
66
  return {"answer": answer}
67
 
68
  if __name__ == "__main__":
69
+ import uvicorn
70
+ uvicorn.run(app, host="0.0.0.0", port=8000)