Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ llama_am = pipeline(
|
|
18 |
)
|
19 |
|
20 |
# Function that accepts a prompt and generates text using the phi2 pipeline
|
21 |
-
def generate(message, chat_history, max_new_tokens=
|
22 |
|
23 |
history = []
|
24 |
|
@@ -38,7 +38,7 @@ def generate(message, chat_history, max_new_tokens=256):
|
|
38 |
kwargs={
|
39 |
"text_inputs":history,
|
40 |
"max_new_tokens":max_new_tokens,
|
41 |
-
"repetition_penalty":1.
|
42 |
"streamer":streamer
|
43 |
}
|
44 |
)
|
@@ -68,6 +68,7 @@ with gr.Blocks() as demo:
|
|
68 |
stop_btn=None,
|
69 |
cache_examples=False,
|
70 |
examples=[
|
|
|
71 |
["የኢትዮጵያ ዋና ከተማ ስም ምንድን ነው?"],
|
72 |
["የኢትዮጵያ የመጨረሻው ንጉስ ማን ነበሩ?"],
|
73 |
["የፈረንሳይ ዋና ከተማ ስም ምንድን ነው?"],
|
@@ -75,7 +76,7 @@ with gr.Blocks() as demo:
|
|
75 |
["ሶስት የአፍሪካ ሀገራት ጥቀስልኝ"],
|
76 |
["3 የአሜሪካ መሪዎችን ስም ጥቀስ"],
|
77 |
["5 የአሜሪካ ከተማዎችን ጥቀስ"],
|
78 |
-
["አምስት የአውሮፓ
|
79 |
["የኢትዮጵያ ፕሬዝዳንት ማን ነው?"],
|
80 |
["በ ዓለም ላይ ያሉትን 7 አህጉራት ንገረኝ"]
|
81 |
]
|
|
|
18 |
)
|
19 |
|
20 |
# Function that accepts a prompt and generates text using the phi2 pipeline
|
21 |
+
def generate(message, chat_history, max_new_tokens=64):
|
22 |
|
23 |
history = []
|
24 |
|
|
|
38 |
kwargs={
|
39 |
"text_inputs":history,
|
40 |
"max_new_tokens":max_new_tokens,
|
41 |
+
"repetition_penalty":1.15,
|
42 |
"streamer":streamer
|
43 |
}
|
44 |
)
|
|
|
68 |
stop_btn=None,
|
69 |
cache_examples=False,
|
70 |
examples=[
|
71 |
+
["ሰላም፣ እንዴት ነህ?"],
|
72 |
["የኢትዮጵያ ዋና ከተማ ስም ምንድን ነው?"],
|
73 |
["የኢትዮጵያ የመጨረሻው ንጉስ ማን ነበሩ?"],
|
74 |
["የፈረንሳይ ዋና ከተማ ስም ምንድን ነው?"],
|
|
|
76 |
["ሶስት የአፍሪካ ሀገራት ጥቀስልኝ"],
|
77 |
["3 የአሜሪካ መሪዎችን ስም ጥቀስ"],
|
78 |
["5 የአሜሪካ ከተማዎችን ጥቀስ"],
|
79 |
+
["አምስት የአውሮፓ ሀገሮችን ጥቀስልኝ"],
|
80 |
["የኢትዮጵያ ፕሬዝዳንት ማን ነው?"],
|
81 |
["በ ዓለም ላይ ያሉትን 7 አህጉራት ንገረኝ"]
|
82 |
]
|