IinjyI commited on
Commit
c469ecf
·
verified ·
1 Parent(s): a9cf0ea

Update app.py

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