Sabbir772 commited on
Commit
22a2ad6
·
verified ·
1 Parent(s): 0831e65

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -0
app.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import AutoModelForCausalLM, AutoTokenizer
3
+ from peft import PeftModel
4
+
5
+ # Load base and adapter model
6
+ base_model = "microsoft/phi-2"
7
+ adapter_model = "Sabbir772/phi2_sylhet"
8
+
9
+ tokenizer = AutoTokenizer.from_pretrained(base_model)
10
+ base = AutoModelForCausalLM.from_pretrained(base_model)
11
+ model = PeftModel.from_pretrained(base, adapter_model)
12
+ model.eval()
13
+
14
+ def infer(text):
15
+ inputs = tokenizer(text, return_tensors="pt").to(model.device)
16
+ outputs = model.generate(**inputs, max_new_tokens=50)
17
+ return tokenizer.decode(outputs[0], skip_special_tokens=True)
18
+
19
+ demo = gr.Interface(fn=infer, inputs="text", outputs="text", title="Phi-2 Sylheti Translator")
20
+ demo.launch()