Update app.py
Browse files
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:
|