Hugging Face's logo

Spaces:
X-iZhang
/
Libra
Running

App Files Community
1
X-iZhang commited on
Commit
ca47d91
·
verified ·
1 Parent(s): 0aeb285

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -43,7 +43,8 @@ def generate_radiology_description(
43
  length_penalty=1.0,
44
  num_return_sequences=1,
45
  conv_mode=conv_mode,
46
- max_new_tokens=max_new_tokens
 
47
  )
48
  return output
49
  except Exception as e:
 
43
  length_penalty=1.0,
44
  num_return_sequences=1,
45
  conv_mode=conv_mode,
46
+ max_new_tokens=max_new_tokens,
47
+ device="cpu"
48
  )
49
  return output
50
  except Exception as e: