liang-huggingface commited on
Commit
c9e7b23
·
1 Parent(s): 29b2b5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
11
  import torch
12
 
13
  # Load the model and tokenizer
14
- tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B-Chat")
15
  model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat")
16
 
17
  def generate_summary(prompt):
 
11
  import torch
12
 
13
  # Load the model and tokenizer
14
+ tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B")
15
  model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat")
16
 
17
  def generate_summary(prompt):