ToletiSri commited on
Commit
b494082
·
verified ·
1 Parent(s): cf5ab2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ model_name = "microsoft/phi-2"
25
  phi2_text = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
26
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
27
 
28
- def example_inference(input_text, count) #, image, img_qn, audio):
29
  pred_text = textMode(input_text, count)
30
  return pred_text #, "in progress", "in progress"
31
 
 
25
  phi2_text = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
26
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
27
 
28
+ def example_inference(input_text, count): #, image, img_qn, audio):
29
  pred_text = textMode(input_text, count)
30
  return pred_text #, "in progress", "in progress"
31