File size: 582 Bytes
c684a3f
 
 
051bdea
 
59db64f
e1090b1
cd8812a
 
00b3c6f
c684a3f
 
 
 
 
 
 
13a7f5c
c684a3f
cd8812a
 
 
 
 
051bdea
c684a3f
9cd77c1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
""" App
"""

import gradio as gr
import time
import os
import elasticsearch
import torch
import torch_xla.core.xla_model as xm

MB = 1024*1024
def eat_memory():
    mem = []
    while True:
        mem += ['c' * (100 * MB)]
        time.sleep(0.1)

#bidule
def greet(name):
    device = xm.xla_device()
    t1 = torch.randn(3,3,device=device)
    t2 = torch.randn(3,3,device=device)
    t3 = str(t1 + t2)
    return "Hello " + name + "!! Tensor: " + t3 + ", secret: " + os.environ.get("TEST_ENV_SECRET")

iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()