Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ tokenizer = BioGptTokenizer.from_pretrained("microsoft/biogpt")
|
|
| 8 |
def get_recommedation(prompt):
|
| 9 |
generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
| 10 |
set_seed(42)
|
| 11 |
-
output = generator(prompt, max_length=
|
| 12 |
return output
|
| 13 |
|
| 14 |
iface = gr.Interface(fn=get_recommedation, inputs="text", outputs="text")
|
|
|
|
| 8 |
def get_recommedation(prompt):
|
| 9 |
generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
| 10 |
set_seed(42)
|
| 11 |
+
output = generator(prompt, max_length=200, num_return_sequences=5, do_sample=True)
|
| 12 |
return output
|
| 13 |
|
| 14 |
iface = gr.Interface(fn=get_recommedation, inputs="text", outputs="text")
|