Chris4K commited on
Commit
0f0577d
1 Parent(s): d833c15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -72,3 +72,13 @@ fact_extraction_chain = LLMChain(llm=llm_hf, prompt=fact_extraction_prompt)
72
  facts = fact_extraction_chain.run(text + " " +output_question_1)
73
 
74
  print(facts)
 
 
 
 
 
 
 
 
 
 
 
72
  facts = fact_extraction_chain.run(text + " " +output_question_1)
73
 
74
  print(facts)
75
+
76
+
77
+ def func (message):
78
+ inputs = tokenizer(message, return_tensors="pt")
79
+ result = model.generate(**inputs)
80
+ return tokenizer.decode(result[0])
81
+
82
+ import gradio as gr
83
+ app = gr.Interface(fn=func, inputs="textbox", outputs="textbox", title="Conversation Bot")
84
+ app.launch()