Update app.py

app.py

@@ -31,7 +31,7 @@ examples = [
         64, "sampling", False],
     ['A poem about the beauty of science by Alfred Edgar Brittle\nTitle: The Magic Craft\nIn the old times', 64,
      "sampling", False],
-    ['استخراج العدد العاملي في لغة بايثون:', 64, "greedy", False],
+    ['استخراج العدد العاملي في لغة بايثون:', 64, "sampling", False],
     ["Pour déguster un ortolan, il faut tout d'abord", 64, "sampling", False],
     [
         'Traduce español de España a español de Argentina\nEl coche es rojo - el auto es rojo\nEl ordenador es nuevo - la computadora es nueva\nel boligrafo es negro -',
@@ -40,13 +40,13 @@ examples = [
         'Estos ejemplos quitan vocales de las palabras\nEjemplos:\nhola - hl\nmanzana - mnzn\npapas - pps\nalacran - lcrn\npapa -',
         64, "sampling", False],
     ["Question: If I put cheese into the fridge, will it melt?\nAnswer:", 64, "sampling", False],
-    ["Math exercise - answers:\n34+10=44\n54+20=", 64, "greedy", False],
+    ["Math exercise - answers:\n34+10=44\n54+20=", 64, "sampling", False],
     [
         "Question: Where does the Greek Goddess Persephone spend half of the year when she is not with her mother?\nAnswer:",
-        64, "greedy", False],
+        64, "sampling", False],
     [
         "spelling test answers.\nWhat are the letters in « language »?\nAnswer: l-a-n-g-u-a-g-e\nWhat are the letters in « Romanian »?\nAnswer:",
-        64, "greedy", False]
+        64, "sampling", False]
 ]
 
 
@@ -69,7 +69,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
     )
 
     if raw_text:
-        return None, data['generated_text']
+        return None, data[0]['generated_text']
 
     width, height = 3326, 3326
     assets_path = "assets"
@@ -87,7 +87,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
         BOX_COMP: "#120F25",
     }
 
-    new_string = data['generated_text']
+    new_string = data[0]['generated_text']
 
     _, img = main(
         input_sentence,
@@ -102,7 +102,7 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
         init_font_size=150,
         right_align=False,
     )
-    return img, data['generated_text']
+    return img, data[0]['generated_text']
 
 
 gr.Interface(