Hugging Face's logo

Spaces:
Sahibsingh12
/
cazton-phi
Runtime error

App Files Community
Sahibsingh12 commited on
Commit
aa3b32f
·
1 Parent(s): f375db4

files changed

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -2,11 +2,8 @@ import gradio as gr
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
- model =
6
- AutoModelForCausalLM.from_pretrained("Sahibsingh12/phi-1-5-finetuned-cazton_complete",
7
- trust_remote_code=True, torch_dtype=torch.float32)
8
- tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5",
9
- trust_remote_code=True)
10
 
11
 
12
  def greet(name):
 
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
+ model = AutoModelForCausalLM.from_pretrained("Sahibsingh12/phi-1-5-finetuned-cazton_complete", trust_remote_code=True, torch_dtype=torch.float32)
6
+ tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5", trust_remote_code=True)
 
 
 
7
 
8
 
9
  def greet(name):