lamini-playground

app.py

@@ -2,8 +2,11 @@ import gradio as gr
 from llama import Type, Context, LLM
 import os
 
-class UserQuery(Type):
-    question: str = Context("what is the user's intent")
+class Question(Type):
+    question: str = Context("a question")
+
+class Response(Type):
+    response: str = Context("the response to the question")
 
 def lamini(input):
     #return "Hello " + name + "!!"
@@ -13,13 +16,13 @@ def lamini(input):
                 "key": os.environ["LAMINI-KEY"]
             }
         })
-    user_query_text=UserQuery(question=input)
+    user_query_text=Question(question=input)
     result = llm(
             input=user_query_text,
-            output_type=UserQuery,
+            output_type=Response,
             model_name="lamini/instruct-tuned-2.8b"
         )
-    return result.question
+    return result.response
 
 iface = gr.Interface(fn=lamini, inputs="text", outputs="text")
 iface.launch()