Update app.py

app.py

@@ -2,10 +2,11 @@ import gradio as gr
 from share_btn import community_icon_html, loading_icon_html, share_js
 import re
 import os 
-hf_token = os.environ.get('HF_TOKEN')
+#hf_token = os.environ.get('HF_TOKEN')
+import openai
+OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
 from gradio_client import Client
-client = Client("https://fffiloni-test-llama-api-debug.hf.space/", hf_token=hf_token)
-
+#client = Client("https://fffiloni-test-llama-api-debug.hf.space/", hf_token=hf_token)
 clipi_client = Client("https://fffiloni-clip-interrogator-2.hf.space/")
 
 def get_text_after_colon(input_text):
@@ -39,12 +40,14 @@ def infer(image_input, audience):
     
     """
     gr.Info('Calling Llama2 ...')
-    result = client.predict(
-    				llama_q,	# str in 'Message' Textbox component
-                    "I2S",
-    				api_name="/predict"
-    )
-
+    #result = client.predict(
+    #				llama_q,	# str in 'Message' Textbox component
+    #                "I2S",
+    #				api_name="/predict"
+    #)
+    chat_completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": llama_q}])
+    result = chat_completion.choices[0].message.content
+    
     print(f"Llama2 result: {result}")
 
     result = get_text_after_colon(result)