Update app.py
Browse files
app.py
CHANGED
@@ -2,12 +2,6 @@ import gradio as gr
|
|
2 |
from transformers import pipeline
|
3 |
model = pipeline("text-generation", model="DmitryYarov/aristotle_based_on_rugpt3large_based_on_gpt")
|
4 |
def predict(prompt):
|
5 |
-
generation_parameters = {
|
6 |
-
"max_length": 500, # Increase the length of the generated text
|
7 |
-
"temperature": 0.9, # Adjust temperature for more creative outputs
|
8 |
-
"top_p": 0.95, # Use nucleus sampling for diversity
|
9 |
-
"num_return_sequences": 1 # Generate one response
|
10 |
-
}
|
11 |
completion = model(prompt)[0]["generated_text"]
|
12 |
return completion
|
13 |
|
|
|
2 |
from transformers import pipeline
|
3 |
model = pipeline("text-generation", model="DmitryYarov/aristotle_based_on_rugpt3large_based_on_gpt")
|
4 |
def predict(prompt):
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
completion = model(prompt)[0]["generated_text"]
|
6 |
return completion
|
7 |
|