krish10 commited on
Commit
d23e6aa
·
verified ·
1 Parent(s): ef2bd77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  import threading
6
 
7
  # Load model and tokenizer
8
- model_name = "krish10/Qwen3_0.6B_16bit_study_design_screen"
9
  pipe = pipeline("text-generation", model=model_name, device=0)
10
  tokenizer = pipe.tokenizer
11
  model = pipe.model
 
5
  import threading
6
 
7
  # Load model and tokenizer
8
+ model_name = "krish10/Qwen2.5_14B_16bit_Study_design_screen"
9
  pipe = pipeline("text-generation", model=model_name, device=0)
10
  tokenizer = pipe.tokenizer
11
  model = pipe.model