Spaces:
Running
Running
update main
Browse files
main.py
CHANGED
|
@@ -93,7 +93,7 @@ def index(req:Request):
|
|
| 93 |
def index(req:Request):
|
| 94 |
return templates.TemplateResponse('ImageInterpretation.html',{'request':req})
|
| 95 |
|
| 96 |
-
app.post("/caption2")
|
| 97 |
async def generate_caption(file: UploadFile = File(...)):
|
| 98 |
contents = await file.read()
|
| 99 |
image = Image.open(io.BytesIO(contents)).convert("RGB")
|
|
@@ -104,7 +104,15 @@ async def generate_caption(file: UploadFile = File(...)):
|
|
| 104 |
caption = interpreter_tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
| 105 |
|
| 106 |
return {"caption": caption}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 107 |
|
|
|
|
|
|
|
| 108 |
@app.post("/caption")
|
| 109 |
def caption(file:UploadFile=File(...)):
|
| 110 |
extension = file.filename.split(".")[-1]
|
|
|
|
| 93 |
def index(req:Request):
|
| 94 |
return templates.TemplateResponse('ImageInterpretation.html',{'request':req})
|
| 95 |
|
| 96 |
+
@app.post("/caption2")
|
| 97 |
async def generate_caption(file: UploadFile = File(...)):
|
| 98 |
contents = await file.read()
|
| 99 |
image = Image.open(io.BytesIO(contents)).convert("RGB")
|
|
|
|
| 104 |
caption = interpreter_tokenizer.decode(output_ids[0], skip_special_tokens=True)
|
| 105 |
|
| 106 |
return {"caption": caption}
|
| 107 |
+
@app.post('/get')
|
| 108 |
+
def g(f:str):
|
| 109 |
+
return generator(f)[0]["generated_text"]
|
| 110 |
+
@app.post('/gets')
|
| 111 |
+
def g(f:str):
|
| 112 |
+
return summarizer(f)[0]['summary_text']
|
| 113 |
|
| 114 |
+
|
| 115 |
+
|
| 116 |
@app.post("/caption")
|
| 117 |
def caption(file:UploadFile=File(...)):
|
| 118 |
extension = file.filename.split(".")[-1]
|