| from transformers import DistilBertTokenizer, DistilBertModel | |
| import torch | |
| tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-cased-distilled-squad') | |
| model = DistilBertModel.from_pretrained('distilbert-base-cased-distilled-squad') | |
| question, text = "Who was Jim Henson?", "Jim Henson was a nice puppet" | |
| inputs = tokenizer(question, text, return_tensors="pt") | |
| with torch.no_grad(): | |
| outputs = model(**inputs) | |
| print(outputs) | |
| import gradio as gr | |
| def greet(name): | |
| return "Hello " + name + "!!" | |
| iface = gr.Interface(fn=greet, inputs="text", outputs="text") | |
| iface.launch() |