Spaces:

greghmerrill
/

gorilla-test2

Runtime error

gmerrill commited on Mar 9, 2024

Commit

8ce0ba9

1 Parent(s): 759408c

update

Files changed (1) hide show

main.py CHANGED Viewed

@@ -1,9 +1,9 @@
 from fastapi import FastAPI, Request
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
 import json
-from transformers import pipeline
 def get_prompt(user_query: str, functions: list = []) -> str:
     """

 from fastapi import FastAPI, Request
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import json
+import torch
 def get_prompt(user_query: str, functions: list = []) -> str:
     """