anmolmalik01 commited on
Commit
03f7088
1 Parent(s): ac3e582

adding files

Browse files
Files changed (2) hide show
  1. __pycache__/main.cpython-310.pyc +0 -0
  2. main.py +1 -3
__pycache__/main.cpython-310.pyc DELETED
Binary file (920 Bytes)
 
main.py CHANGED
@@ -1,12 +1,10 @@
1
  from fastapi import FastAPI
2
  from fastapi.staticfiles import StaticFiles
3
  from fastapi.responses import FileResponse
4
-
5
  from transformers import pipeline
6
 
7
- from pydantic import BaseModel
8
-
9
  app = FastAPI()
 
10
  summarizer = pipeline("summarization", model="Falconsai/text_summarization")
11
 
12
  @app.get("/prediction")
 
1
  from fastapi import FastAPI
2
  from fastapi.staticfiles import StaticFiles
3
  from fastapi.responses import FileResponse
 
4
  from transformers import pipeline
5
 
 
 
6
  app = FastAPI()
7
+
8
  summarizer = pipeline("summarization", model="Falconsai/text_summarization")
9
 
10
  @app.get("/prediction")