Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
import pickle
|
3 |
import zipfile
|
@@ -125,7 +126,21 @@ def translate(text):
|
|
125 |
out = model.generate(**tokenized)
|
126 |
arabic = tokenizer.decode(out[0], skip_special_tokens=True)
|
127 |
return arabic
|
128 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
129 |
demo = gr.Interface(fn=translate_pretrained, inputs="text", outputs="text")
|
130 |
demo.launch(share=True)
|
131 |
|
|
|
1 |
+
from fastapi import FastAPI, File, UploadFile
|
2 |
import gradio as gr
|
3 |
import pickle
|
4 |
import zipfile
|
|
|
126 |
out = model.generate(**tokenized)
|
127 |
arabic = tokenizer.decode(out[0], skip_special_tokens=True)
|
128 |
return arabic
|
129 |
+
|
130 |
+
app = FastAPI()
|
131 |
+
|
132 |
+
|
133 |
+
@app.post("/text/")
|
134 |
+
def create_text(text: str):
|
135 |
+
return {"summary": translate_pretrained(text)}
|
136 |
+
|
137 |
+
|
138 |
+
@app.post("/file/")
|
139 |
+
def create_upload_file(file: UploadFile = File(...)):
|
140 |
+
text = file.file.read().decode("utf-8")
|
141 |
+
return {"summary": translate_pretrained(text)}
|
142 |
+
|
143 |
+
|
144 |
demo = gr.Interface(fn=translate_pretrained, inputs="text", outputs="text")
|
145 |
demo.launch(share=True)
|
146 |
|