Tonic commited on
Commit
2469625
1 Parent(s): afd1412

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def multimodal_prompt(user_input, system_prompt="You are an expert medical analy
29
  model_inputs = encodeds.to(device)
30
 
31
  # Generate a response using the model
32
- output = model.generate(
33
  **model_inputs,
34
  max_length=500,
35
  use_cache=True,
 
29
  model_inputs = encodeds.to(device)
30
 
31
  # Generate a response using the model
32
+ output = peft_model.generate(
33
  **model_inputs,
34
  max_length=500,
35
  use_cache=True,