rajivmehtapy commited on
Commit
04fc58a
·
1 Parent(s): a9b2acf

remove gpu requirements.

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -35,12 +35,12 @@ llm = LlamaCpp(
35
  )
36
  llm_chain = prompt | llm
37
 
38
- @spaces.GPU(duration=120)
39
  def greet(name):
40
- logger.info("This is an info message")
41
  question = name
42
- # print(llm_chain.invoke({"question": question}))
43
- return llm_chain.invoke({"question": question})
 
44
 
45
  demo = gr.Interface(fn=greet, inputs="text", outputs="text")
46
  demo.launch()
 
35
  )
36
  llm_chain = prompt | llm
37
 
38
+ # @spaces.GPU(duration=120)
39
  def greet(name):
 
40
  question = name
41
+ response = llm_chain.invoke({"question": question})
42
+ logger.info(f"Response --> {response}")
43
+ return
44
 
45
  demo = gr.Interface(fn=greet, inputs="text", outputs="text")
46
  demo.launch()