mipra commited on
Commit
b23f98e
β€’
1 Parent(s): f3a7389

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -1,15 +1,24 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 
 
 
 
 
 
3
 
4
  def translate(text):
5
  inputs = tokenizer.encode(
6
  hin_snippet, return_tensors="pt",padding=True,max_length=512,truncation=True)
7
  outputs = model.generate(
8
  inputs, max_length=128, num_beams=None, early_stopping=True)
 
9
 
10
  translated = tokenizer.decode(outputs[0]).replace('<pad>',"").strip().lower()
11
-
12
- return translated
 
13
 
14
 
15
  tokenizer = AutoTokenizer.from_pretrained("salesken/translation-hi-en")
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
+ import requests
4
+
5
+
6
+
7
+
8
+
9
+
10
 
11
  def translate(text):
12
  inputs = tokenizer.encode(
13
  hin_snippet, return_tensors="pt",padding=True,max_length=512,truncation=True)
14
  outputs = model.generate(
15
  inputs, max_length=128, num_beams=None, early_stopping=True)
16
+
17
 
18
  translated = tokenizer.decode(outputs[0]).replace('<pad>',"").strip().lower()
19
+ r = requests.post(url='https://hf.space/embed/multimodalart/latentdiffusion/+/api/predict/',
20
+ json={"data":[translated,50,32,32,4,15.0]})
21
+ return r.json()
22
 
23
 
24
  tokenizer = AutoTokenizer.from_pretrained("salesken/translation-hi-en")