Shanat commited on
Commit
c315db2
·
verified ·
1 Parent(s): f91bbe8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ from transformers import pipeline
10
  import os
11
  from huggingface_hub import login
12
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
13
 
14
 
15
  #chatbot = pipeline(model="NCSOFT/Llama-3-OffsetBias-8B")
 
10
  import os
11
  from huggingface_hub import login
12
  from transformers import AutoModelForCausalLM, AutoTokenizer
13
+ import torch
14
 
15
 
16
  #chatbot = pipeline(model="NCSOFT/Llama-3-OffsetBias-8B")