Johnyquest7 commited on
Commit
2961e7c
1 Parent(s): 6671a63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import spaces
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
6
 
7
 
8
- TITLE = "<h1><center>Chat with models/meta-llama/Meta-Llama-3-8B-Instruct</center></h1>"
9
 
10
  DESCRIPTION = "<h3><center>An experimental space.</center></h3>"
11
 
@@ -102,8 +102,8 @@ CSS = """
102
  """
103
 
104
 
105
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
106
- model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", torch_dtype="auto", device_map="auto")
107
 
108
 
109
  @spaces.GPU
 
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
6
 
7
 
8
+ TITLE = "<h1><center>Chat with microsoft/Phi-3-mini-4k-instructt</center></h1>"
9
 
10
  DESCRIPTION = "<h3><center>An experimental space.</center></h3>"
11
 
 
102
  """
103
 
104
 
105
+ tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
106
+ model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", torch_dtype="auto", device_map="auto")
107
 
108
 
109
  @spaces.GPU