Update app.py

app.py

@@ -5,7 +5,7 @@ prompt1=''' is there any code Snippets present in the below text? Reply with YES
 
 Text: " {} "
 '''
-prompt2='''Extract Code Snippets ONLY from the below text .This is very important for my career:
+prompt2='''Extract Code Snippets ONLY.This is very important for my career:
 
 Text: " {} "
 '''
@@ -29,16 +29,17 @@ def chatbot_model(m,history):
     decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
     if (decoded_output=='NO'):
         yield 'NO'
-    input_ids = tokenizer.encode(prompt2.format(m) + tokenizer.eos_token, return_tensors='pt')
-    outputs = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
+    else:
+        input_ids = tokenizer.encode(prompt2.format(m) + tokenizer.eos_token, return_tensors='pt')
+        outputs = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
     
     # Decode the output
-    decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    for char, replacement in replacements.items():
-        decoded_output = decoded_output.replace(replacement, char)    
-    for i in range(len(decoded_output)):
-        time.sleep(0.3)
-        yield decoded_output[:i+1]
+        decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        for char, replacement in replacements.items():
+            decoded_output = decoded_output.replace(replacement, char)    
+        for i in range(len(decoded_output)):
+            time.sleep(0.3)
+            yield decoded_output[:i+1]
 
 iface = gr.ChatInterface(chatbot_model,title="Code Detector",undo_btn=None,theme="soft").queue()
 iface.launch()