AminFaraji commited on
Commit
4ecbcef
·
verified ·
1 Parent(s): 9614f6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ llama_pipeline = pipeline(
57
  )
58
 
59
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
60
- print(f"Model device: {model.device}")
61
 
62
  from transformers import AutoModel,AutoTokenizer
63
  model2 = AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
 
57
  )
58
 
59
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
60
+
61
 
62
  from transformers import AutoModel,AutoTokenizer
63
  model2 = AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")