add opus

README.md

@@ -1,8 +1,8 @@
 ---
-title: Mariobedrock
-emoji: 🐨
-colorFrom: purple
-colorTo: gray
+title: MarioBedrock - Generate playable Super Mario levels from a prompt!
+emoji: 🍄
+colorFrom: red
+colorTo: green
 sdk: gradio
 sdk_version: 4.27.0
 app_file: app.py

app.py

@@ -10,6 +10,10 @@ import uvicorn
 import boto3
 import json
 
+import anthropic
+
+opus_client = anthropic.Anthropic()
+
 bedrock_runtime = boto3.client(
     service_name="bedrock-runtime",
     region_name="us-east-1",
@@ -143,6 +147,19 @@ def call_llama3_8b(system_prompt, prompt, temperature):
     return results
 
 
+def call_claude_3_opus(system_prompt, prompt, temperature):
+
+    message = opus_client.messages.create(
+        model="claude-3-opus-20240229",
+        max_tokens=4096,
+        system=system_prompt,
+        temperature=temperature,
+        messages=[{"role": "user", "content": prompt}],
+    )
+
+    return message.content[0].text
+
+
 # def call_claude_3_opus(system_prompt, prompt, temperature):
 
 #     prompt_config = {
@@ -365,8 +382,8 @@ def generate(model, prompt, temperature, system_prompt=system_prompt_text):
         level = call_llama3_70b(system_prompt, prompt, temperature)
     elif model == "Llama3 8B":
         level = call_llama3_8b(system_prompt, prompt, temperature)
-    # elif model == "Cladue Opus":
-    #     level = call_claude_3_opus(system_prompt, prompt, temperature)
+    elif model == "Claude Opus":
+        level = call_claude_3_opus(system_prompt, prompt, temperature)
     else:
         raise ValueError("Invalid model")
 
@@ -418,14 +435,14 @@ You can try it out by entering in a prompt and clicking `Generate level` to play
 
             model = gr.Radio(
                 [
-                    # "Cladue Opus", # no opus for demo
+                    "Claude Opus",
                     "Claude Sonnet",
                     "Claude Haiku",
                     "Llama3 70B",
                     "Llama3 8B",
                 ],
                 label="Select Model",
-                value="Claude Sonnet",
+                value="Claude Opus",
             )
 
     with gr.Accordion(label="Advanced settings", open=False):
@@ -463,7 +480,7 @@ You can try it out by entering in a prompt and clicking `Generate level` to play
     gr.Examples(
         examples=[
             [
-                "Claude Sonnet",
+                "Claude Opus",
                 "Generate a fun level, make sure Mario will have a good time!!!",
                 1.0,
             ],

requirements.txt

@@ -7,4 +7,5 @@ fastapi
 gradio
 boto3
 torch
-accelerate
+accelerate
+anthropic