Dejansimic
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -2,10 +2,10 @@ import gradio as gr
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
# Inicijalizuj model za generisanje teksta
|
5 |
-
text_generator = pipeline("text-generation", model="
|
6 |
|
7 |
def generate_blog_post(topic):
|
8 |
-
# Generiši tekst na osnovu
|
9 |
response = text_generator(topic, max_length=300, num_return_sequences=1)
|
10 |
return response[0]['generated_text']
|
11 |
|
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
# Inicijalizuj model za generisanje teksta
|
5 |
+
text_generator = pipeline("text-generation", model="nvidia/Llama-3.1-Nemotron-70B-Instruct-HF")
|
6 |
|
7 |
def generate_blog_post(topic):
|
8 |
+
# Generiši tekst na osnovu zadate teme
|
9 |
response = text_generator(topic, max_length=300, num_return_sequences=1)
|
10 |
return response[0]['generated_text']
|
11 |
|