changes to app

app.py

@@ -1,13 +1,13 @@
+import pyjokes
+import gradio as gr
+import numpy as np
+from nltk.corpus import wordnet as wn
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
 
 import nltk
 nltk.download('all')
-from nltk.corpus import wordnet as wn
-import numpy as np
 
-import gradio as gr
-import pyjokes
 
 def similarity(input, joke):
     return cosine_similarity(input, joke)
@@ -21,7 +21,8 @@ def get_best(input):
     jokes_embedding = model.encode(jokes)
     input_embedding = model.encode(input)
     for idx, joke_embedding in enumerate(jokes_embedding):
-        sim = similarity(joke_embedding.reshape(-1, 1), input_embedding.reshape(-1,1))
+        sim = similarity(joke_embedding.reshape(-1, 1),
+                         input_embedding.reshape(-1, 1))
         if(np.sum(sim) > np.sum(max_similarity)):
             max_idx = idx
             max_similarity = sim
@@ -30,6 +31,7 @@ def get_best(input):
     else:
         return None
 
+
 def generate_list(input):
     result = []
     n = len(input)
@@ -46,17 +48,18 @@ def generate_list(input):
 def pattern(input):
     response = input
     for substr in generate_list(input):
-        try :
-          syn = wn.synsets(substr)[1].hypernyms()[0].hyponyms()[0].hyponyms()[0].lemmas()[0].name()
+        try:
+            syn = wn.synsets(substr)[1].hypernyms()[0].hyponyms()[
+                0].hyponyms()[0].lemmas()[0].name()
         except:
-          continue
+            continue
         if(syn != None):
             response = response.replace(substr, syn.upper())
 
     if(input == response):
-      return None
-    else :
-      return response
+        return None
+    else:
+        return response
 
 
 def GPT(input):
@@ -90,7 +93,8 @@ def generator(input=None):
         if(out2):
             response.append(out2)
 
-    return response #[0] think of doing this
+    return response[0]  # think of doing this
+
 
 iface = gr.Interface(fn=generator, inputs="text", outputs="text")
-iface.launch()
+iface.launch()