Paridhim commited on
Commit
6ccb460
1 Parent(s): 0e023da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from langchain_community.llms.ctransformers import CTransformers
2
  import os
 
3
 
4
  hf_token = os.environ.get('HF_TOKEN')
5
 
@@ -35,8 +36,8 @@ def generate_text(input_text):
35
 
36
  output = llm(input_text)
37
  print(output)
38
-
39
- return output
40
 
41
 
42
  text_generation_interface = gr.Interface(
 
1
  from langchain_community.llms.ctransformers import CTransformers
2
  import os
3
+ import gradio as gr
4
 
5
  hf_token = os.environ.get('HF_TOKEN')
6
 
 
36
 
37
  output = llm(input_text)
38
  print(output)
39
+ cleaned_output_text = output_text.replace(input_text, "")
40
+ return cleaned_output_text
41
 
42
 
43
  text_generation_interface = gr.Interface(