Hugging Face's logo

Spaces:
midrees2806
/
Llama2_space
Sleeping

App Files Community
midrees2806 commited on
Commit
6d602ad
·
verified ·
1 Parent(s): 1277676

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from transformers import LlamaTokenizer, LlamaForCausalLM, pipeline
3
-
4
  # Load your model and tokenizer
5
  model_name = "midrees2806/Llama-2-7b-updatedchatbot-finetune"
6
  tokenizer = LlamaTokenizer.from_pretrained(model_name)
 
1
  import gradio as gr
2
  from transformers import LlamaTokenizer, LlamaForCausalLM, pipeline
3
+ import torch
4
  # Load your model and tokenizer
5
  model_name = "midrees2806/Llama-2-7b-updatedchatbot-finetune"
6
  tokenizer = LlamaTokenizer.from_pretrained(model_name)