Upload folder using huggingface_hub

.idea/workspace.xml

@@ -32,13 +32,13 @@
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent">{
-  &quot;keyToString&quot;: {
-    &quot;Python.main.executor&quot;: &quot;Run&quot;,
-    &quot;RunOnceActivity.ShowReadmeOnStart&quot;: &quot;true&quot;,
-    &quot;settings.editor.selected.configurable&quot;: &quot;preferences.pluginManager&quot;
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "Python.main.executor": "Run",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "settings.editor.selected.configurable": "preferences.pluginManager"
   }
-}</component>
+}]]></component>
   <component name="RunManager">
     <configuration name="main" type="PythonConfigurationType" factoryName="Python" nameIsGenerated="true">
       <module name="gradioapp" />
@@ -82,6 +82,25 @@
     <servers />
   </component>
   <component name="XDebuggerManager">
+    <breakpoint-manager>
+      <breakpoints>
+        <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
+          <url>file://$PROJECT_DIR$/main.py</url>
+          <line>47</line>
+          <option name="timeStamp" value="2" />
+        </line-breakpoint>
+        <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
+          <url>file://$PROJECT_DIR$/main.py</url>
+          <line>25</line>
+          <option name="timeStamp" value="3" />
+        </line-breakpoint>
+        <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
+          <url>file://$PROJECT_DIR$/main.py</url>
+          <line>49</line>
+          <option name="timeStamp" value="4" />
+        </line-breakpoint>
+      </breakpoints>
+    </breakpoint-manager>
     <watches-manager>
       <configuration name="PythonConfigurationType">
         <watch expression="messages[0][&quot;generated_text&quot;]" language="Python" />

README.md

@@ -4,4 +4,4 @@ app_file: main.py
 sdk: gradio
 sdk_version: 4.36.1
 python_version: 3.11.3
----
+---

main.py

@@ -5,61 +5,73 @@
 
 
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, Pipeline
 from transformers import Conversation
 
-
 def chatwith_blenderbot400m():
-    chatbot = pipeline(task="conversational", model="facebook/blenderbot-400M-distill")
+    pipe = pipeline(task="conversational", model="facebook/blenderbot-400M-distill")
     user_message = "What are some fun activities I can do in the winter?"
     conversation = Conversation(user_message)
     print(conversation)
     print(type(conversation))
-    conversation = chatbot(conversation)
+    conversation = pipe(conversation)
     print(conversation)
     conversation.add_message(
         {"role": "user", "content": "I would like to do outdoor activities. Which activities can I do?"})
-    conversation = chatbot(conversation)
+    conversation = pipe(conversation)
     print(conversation)
 
 
 def chatwith_qwen2_1point5b_instruct():
-    chatbot = pipeline(task="text-generation", model="Qwen/Qwen2-1.5B-Instruct")
+    pipe = pipeline(task="text-generation", model="Qwen/Qwen2-1.5B-Instruct")
     messages = [{"role": "user", "content": "What are some fun activities I can do in the winter?"}]
 
-    messages = chatbot(messages, max_new_tokens=50)[0]["generated_text"]
+    messages = pipe(messages, max_new_tokens=50)[0]["generated_text"]
     print(messages)
 
     messages.append({"role": "user", "content": "I would like to do outdoor activities. Which activities can I do?"})
     print(messages)
-    messages = chatbot(messages, max_new_tokens=50)[0]["generated_text"]
+    messages = pipe(messages, max_new_tokens=50)[0]["generated_text"]
     print(messages)
 
-
+#chatwith_qwen2_1point5b_instruct()
 def chatwith_qwen2_1point5b_instruct(prompt, max_newtokens):
     print("Aaaaa")
-    chatbot = pipeline(task="text-generation", model="Qwen/Qwen2-1.5B-Instruct")
+    pipe = pipeline(task="text-generation", model="Qwen/Qwen2-1.5B-Instruct")
 
     messages = [{"role": "user", "content": prompt}]
-    messages = chatbot(messages, max_new_tokens=max_newtokens)[0]["generated_text"]
+    messages = pipe(messages, max_new_tokens=max_newtokens)[0]["generated_text"]
     return messages
 
+pipe = pipeline(task="text-generation", model="Qwen/Qwen2-1.5B-Instruct")
 
-#chatwith_blenderbot400m()
-#chatwith_qwen2_1point5b_instruct()
 
-# prompt = "What are some fun activities I can do in the winter?"
-# max_newtokens = 2
-# print(chatwith_qwen2_1point5b_instruct(prompt, max_newtokens))
+def chatbot_handler(user_message, history):
+    bot_response = "I don't think so"
+    messages = []
+
+    user_message = {"role": "user", "content": user_message}
+    # TODO: build messages based on history then add user_message to messages. call model
+
+    for message in history:
+        messages.append({"role": "user", "content": message[0]})
+        messages.append({"role": "assistant", "content": message[1]})
+        # print(message[0])
+        # print(message[1])
+    messages.append(user_message)
+
+    print(f"messages before sending to model {messages}")
+    messages = pipe(messages, max_new_tokens=50)[0]['generated_text']
+    print(f"messages after sending to model{messages}")
 
-# def greet(name, intensity):
-#     return "Hello........., " + name + "!" * int(intensity)
+    if messages:
+        # messages has at least one item
+        print(f"the last message is: {messages[-1]}")
+        bot_response = messages[-1]["content"]
+        print(bot_response)
+    return bot_response
 
 
-demo = gr.Interface(
-    fn=chatwith_qwen2_1point5b_instruct,
-    inputs=["text", "slider"],
-    outputs=["text"],
-)
+chatbot = gr.ChatInterface(chatbot_handler)
+chatbot.launch(share=False)
 
-demo.launch(share=False)