ejschwartz commited on
Commit
d4ae7ba
·
1 Parent(s): ac00bbf

Delay creation of gradio client

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,7 +22,7 @@ fielddecoder_model = AutoModelForCausalLM.from_pretrained(
22
  "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16
23
  ).to("cuda")
24
 
25
- gradio_client = Client("https://ejschwartz-resym-field-helper.hf.space/")
26
 
27
  examples = [
28
  ex.encode().decode("unicode_escape") for ex in open("examples.txt", "r").readlines()
@@ -34,7 +34,7 @@ examples = [
34
  # "output": "a1: a, os_reltime* -> sec, os_time_t\na1[1]: a, os_reltime* -> usec, os_time_t\na2: b, os_reltime* -> sec, os_time_t\na2[1]: b, os_reltime* -> usec, os_time_t",
35
  def field_prompt(code):
36
  try:
37
- field_helper_result = gradio_client.predict(
38
  decompiled_code=code,
39
  api_name="/predict",
40
  )
 
22
  "ejschwartz/resym-fielddecoder", torch_dtype=torch.bfloat16
23
  ).to("cuda")
24
 
25
+ make_gradio_client = lambda: Client("https://ejschwartz-resym-field-helper.hf.space/")
26
 
27
  examples = [
28
  ex.encode().decode("unicode_escape") for ex in open("examples.txt", "r").readlines()
 
34
  # "output": "a1: a, os_reltime* -> sec, os_time_t\na1[1]: a, os_reltime* -> usec, os_time_t\na2: b, os_reltime* -> sec, os_time_t\na2[1]: b, os_reltime* -> usec, os_time_t",
35
  def field_prompt(code):
36
  try:
37
+ field_helper_result = make_gradio_client().predict(
38
  decompiled_code=code,
39
  api_name="/predict",
40
  )